MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

↓ 1 callersFunctionDeduceStepPredicate
If `backedge_predicate` is equal to `symbolic_predicate` & Step where Step does not contain `symbolic_predicate` as an inner (not top-level) operand t
tao/tao_bridge/tf/deadness_analysis.cc:1029
↓ 1 callersFunctionDefuse
tao/tao_bridge/passes/tao_defuse_pass_test.cc:203
↓ 1 callersFunctionDefuseActivation
tao/tao_bridge/passes/tao_defuse_pass.cc:120
↓ 1 callersFunctionDefuseBatchNorm
tao/tao_bridge/passes/tao_defuse_pass.cc:140
↓ 1 callersFunctionDefuseBiasAdd
tao/tao_bridge/passes/tao_defuse_pass.cc:100
↓ 1 callersFunctionDefuseConv2D
tao/tao_bridge/passes/tao_defuse_pass.cc:170
↓ 1 callersFunctionDefuseFromRoot
tao/tao_bridge/passes/tao_defuse_pass.cc:207
↓ 1 callersFunctionDefuseMatMul
tao/tao_bridge/passes/tao_defuse_pass.cc:74
↓ 1 callersMethodDeregisterCache
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:356
↓ 1 callersFunctionDeriveInputPlacement
tao_compiler/mlir/disc/tools/disc-replay/record.cc:18
↓ 1 callersFunctionDiscCMD
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/register_disc_class.cpp:38
↓ 1 callersFunctionDumpIValue
pytorch_blade/pytorch_blade/common_utils/utils.cpp:56
↓ 1 callersFunctionDumpProgramAndData
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:146
↓ 1 callersMethodDumpToFile
tao/tao_bridge/kernels/tao_compilation_cache.cc:617
↓ 1 callersFunctionEncapsulateSubgraphsInFunctions
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2540
↓ 1 callersFunctionEndClusterReplayRecord
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:199
↓ 1 callersFunctionEnhancementInputShape
pytorch_blade/pytorch_blade/ltc/disc_compiler/disc_compiler.cpp:81
↓ 1 callersMethodEnqueueAction
tao/tao_bridge/kernels/tao_compilation_cache.cc:172
↓ 1 callersMethodExecute
pytorch_blade/pytorch_blade/compiler/backends/engine_class.cpp:72
↓ 1 callersFunctionExtractConstValue
TODO(disc): refactor this part of code to phase out using HEX format.
tao_compiler/mlir/disc/transforms/disc_convert_const_to_ral.cc:47
↓ 1 callersFunctionFillDescriptors
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:1071
↓ 1 callersFunctionFillStridesAndDilation
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:1044
↓ 1 callersFunctionFilterInvalidClusters
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2088
↓ 1 callersMethodFindClusterDependencies
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2240
↓ 1 callersFunctionFindFirstOf
Like s.find_first_of(x, pos), but return s.size() when find_first_of() would return string::npos. This avoids if-statements elsewhere.
tao/tao_bridge/tf/parse_flags_from_env.cc:80
↓ 1 callersMethodFindLowerToLibraryCallFunc
tao_compiler/mlir/disc/IR/custom_call_base.cc:64
↓ 1 callersFunctionFindNodeByName
tao/tao_bridge/passes/tao_clone_constants_for_better_clustering_test.cc:70
↓ 1 callersMethodFindReifyShapesFunc
tao_compiler/mlir/disc/IR/custom_call_base.cc:53
↓ 1 callersMethodFindRoot
tao/tao_bridge/tf/union_find.h:72
↓ 1 callersMethodFindUncompilableNodes
tao/tao_bridge/tf/compilability_check_util.cc:122
↓ 1 callersFunctionFloorOfRatio
tao/tao_bridge/tf/util.h:382
↓ 1 callersMethodFlushCallTimestamp
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:584
↓ 1 callersMethodFoldAccAlltoAll
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/distributed/kernel/custom.cpp:62
↓ 1 callersMethodFoldAccGather
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/distributed/kernel/custom.cpp:21
↓ 1 callersMethodFoldAccScatter
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/distributed/kernel/custom.cpp:41
↓ 1 callersFunctionFoldOutputs
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:34
↓ 1 callersMethodFunctionExists
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:379
↓ 1 callersFunctionFunctionalizeControlFlow
tao/tao_bridge/passes/functionalize_control_flow.cc:125
↓ 1 callersMethodFunctionalizeInternal
tao/tao_bridge/passes/functionalize_cond.cc:1267
↓ 1 callersFunctionFunctionalizeLoop
tao/tao_bridge/passes/functionalize_while.cc:285
↓ 1 callersFunctionFunctionalizeWhileLoop
tao/tao_bridge/passes/functionalize_while.cc:598
↓ 1 callersFunctionGDef
tao/tao_bridge/tao_util_test.cc:50
↓ 1 callersFunctionGenerateUniqueNameForConst
unique_name is in the format: {hash_of_literal}_2x3x4xf32
tao_compiler/mlir/disc/transforms/disc_convert_const_to_ral.cc:54
↓ 1 callersFunctionGetAlgorithms
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:758
↓ 1 callersMethodGetAncestorId
tao/tao_bridge/passes/functionalize_cond.cc:212
↓ 1 callersMethodGetArgIndexForEdge
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:723
↓ 1 callersFunctionGetArgTypes
Finds the types of the _Arg nodes, indexed by position.
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2565
↓ 1 callersFunctionGetArrayOfElementPointers
tao_compiler/mlir/xla/ral/context/custom_library/gpu_helper.h:38
↓ 1 callersFunctionGetArrayOfElementPointersImpl
tao_compiler/mlir/xla/ral/context/custom_library/gpu_helper.h:32
↓ 1 callersMethodGetCache
tao_compiler/mlir/xla/ral/ral_context.cc:86
↓ 1 callersFunctionGetCandidateColReduceOp
tao_compiler/mlir/disc/transforms/disc_specialize_fusion_with_speculation.cc:200
↓ 1 callersMethodGetClusterName
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:1906
↓ 1 callersMethodGetCompilationDevice
static */
tao/tao_bridge/tf/xla_op_registry.cc:151
↓ 1 callersFunctionGetCompilers
tao_compiler/decoupling/compiler_base.cc:27
↓ 1 callersFunctionGetConvMetadata
tao_compiler/mlir/disc/IR/qconv_custom_call_op.cc:181
↓ 1 callersFunctionGetDISCBackendImpl
pytorch_blade/pytorch_blade/ltc/disc_backend/backend_impl.cpp:283
↓ 1 callersFunctionGetDeviceAndAllocator
tensorflow_blade/src/util/tf_allocator_util.cc:57
↓ 1 callersFunctionGetDeviceId
tao/tao_bridge/kernels/tao_compilation_cache.cc:309
↓ 1 callersMethodGetDeviceTypeFor
tao/tao_bridge/tf/device_util.h:143
↓ 1 callersFunctionGetDiscSupportedOps
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2562
↓ 1 callersFunctionGetF32ElementsAttr
Gets all float values from the given attribute and push them to `values`.
tao_compiler/mlir/disc/disc_util.cc:325
↓ 1 callersFunctionGetFirstValidDeviceId
tensorflow_blade/src/util/tf_allocator_util.cc:42
↓ 1 callersFunctionGetFullFrame
tao/tao_bridge/tf/deadness_analysis.cc:1074
↓ 1 callersMethodGetFuncNum
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:363
↓ 1 callersMethodGetFuncShapeNum
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:369
↓ 1 callersFunctionGetFunctionAttrs
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:99
↓ 1 callersFunctionGetFunctionBodies
tao/tao_bridge/tf/const_analysis.cc:47
↓ 1 callersFunctionGetFusedOpTable
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2556
↓ 1 callersFunctionGetGpuInfo
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:67
↓ 1 callersFunctionGetGpuNumber
tao/tao_bridge/tf/xla_cluster_util.cc:259
↓ 1 callersFunctionGetGpuPCIBusID
tao/tao_bridge/cuda_utils_test.cc:32
↓ 1 callersFunctionGetI64ElementsAttrForSeq
Returns a 1-d i64 elements attribute populated with numbers from start to end, excluding.
tao_compiler/mlir/disc/disc_util.cc:341
↓ 1 callersMethodGetIdOfOnlyNode
If this is a trivial cluster containing only one node then return the ID of that node. May not be called otherwise.
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:174
↓ 1 callersMethodGetLeaderValueWithSameShape
tao_compiler/mlir/disc/transforms/shape_utils.h:390
↓ 1 callersFunctionGetLibdeviceDir
tao_compiler/mlir/disc/transforms/legalize_to_cubin.cc:97
↓ 1 callersFunctionGetLinkedTensorRTVersion
tensorflow_blade/src/tensorrt/bridge/tensorrt_common.cpp:24
↓ 1 callersFunctionGetLinkedTensorRTVersion
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_common.cpp:24
↓ 1 callersFunctionGetLoadedTensorRTVersion
tensorflow_blade/src/tensorrt/bridge/tensorrt_common.cpp:31
↓ 1 callersFunctionGetLoadedTensorRTVersion
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_common.cpp:31
↓ 1 callersFunctionGetNextClusterSequenceNumber
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:915
↓ 1 callersMethodGetOrCreateCacheIdx
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:336
↓ 1 callersFunctionGetOrCreateDefaultCudaStreamExecutorStream
tao_compiler/mlir/xla/ral/context/init_stream_executor.cc:56
↓ 1 callersFunctionGetOrCreateWhitelist
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:1213
↓ 1 callersFunctionGetOutputOp
Retrieves the lowered operation for an output. If the requested output is not available yet, the graph behind the output's Node is lowered, and the co
pytorch_blade/pytorch_blade/ltc/include/torch/csrc/lazy/ts_backend/ts_lowering_context.h:126
↓ 1 callersMethodGetPredicateFor
tao/tao_bridge/tf/deadness_analysis.cc:1555
↓ 1 callersMethodGetRecvAtHostSlot
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:737
↓ 1 callersMethodGetResultIndexForEdge
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:727
↓ 1 callersFunctionGetRetTypes
Finds the types of the _Retval nodes, indexed by position.
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2597
↓ 1 callersFunctionGetRngUniqueId
tao_compiler/mlir/disc/IR/rng_uniform_custom_call_op.cc:85
↓ 1 callersFunctionGetScalarOfType
tao_compiler/mlir/disc/disc_util.h:79
↓ 1 callersFunctionGetSegmentList
tensorflow_blade/src/tensorrt/bridge/tensorrt_onnx_parser.cpp:27
↓ 1 callersFunctionGetSegmentList
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_onnx_parser.cpp:21
↓ 1 callersMethodGetSendFromHostSlot
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:749
↓ 1 callersMethodGetShapeCallCount
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:473
↓ 1 callersMethodGetShapeString
pytorch_blade/pytorch_blade/compiler/backends/backend_input_outputs.cpp:73
↓ 1 callersFunctionGetTargetMachine
tao_compiler/mlir/disc/disc_compiler.cc:131
↓ 1 callersFunctionGetTempDirectory
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/io.h:29
↓ 1 callersFunctionGetTensorRTBindingTypes
tensorflow_blade/src/tensorrt/pybind_functions.cpp:35
↓ 1 callersFunctionGetTestSeed
Return a test seed.
tao_compiler/mlir/xla/ral/context/custom_library/philox_random_test.cc:39
↓ 1 callersFunctionGetTestingRecord
tao_compiler/mlir/disc/tools/disc-replay/replay_test.cc:32
← previousnext →1,401–1,500 of 6,088, ranked by callers