MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

↓ 2 callersFunctiontryToExtractDimMap
tao_compiler/mlir/disc/transforms/disc_algebraic_simplifier.cc:245
↓ 2 callersFunctiontry_cast_to_float
(to_cast: str)
pytorch_blade/benchmark/TorchBench/results_analysis.py:34
↓ 2 callersFunctionunionScalarTypes
pytorch_blade/pytorch_blade/compiler/jit/torch/shape_analysis.cpp:302
↓ 2 callersMethodunregisterConstStore
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:157
↓ 2 callersMethodupdateKnownNonSizeZero
tao_compiler/mlir/disc/IR/disc_shape_ops.cc:352
↓ 2 callersFunctionupdate_cpu_specific_setting
(args)
scripts/python/common_setup.py:373
↓ 2 callersFunctionupper
()
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/utils.py:184
↓ 2 callersMethodvalue2SymbolicDimOp
tao_compiler/mlir/disc/transforms/disc_shape_optimization.cc:1446
↓ 2 callersFunctionversion_info
tao_compiler/decoupling/tao_compiler_main.cc:64
↓ 2 callersFunctionwarpReduceMax
tao_compiler/mlir/xla/ral/context/custom_library/warp.cu.h:31
↓ 2 callersFunctionwithInputFromWhereOp
NOTE(lanbo.llb): mhlo.disc op will produce 2 outputs, the 2nd output is i64 tensor with single element represent number of valid elements in the 1st o
tao_compiler/mlir/disc/transforms/mhlo_mark_shape_calc.cc:99
↓ 1 callersMethodActivateContext
tao_compiler/mlir/xla/ral/context/init_stream_executor.cc:33
↓ 1 callersMethodAddCompileFailedCase
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:998
↓ 1 callersFunctionAddDummyShapedNode
Adds a dummy Const node to graph_out. The "constant" has the type of data_type and the shape indicated in 'shape'. The dummy node is not a valid Const
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1898
↓ 1 callersMethodAddFunction
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:392
↓ 1 callersMethodAddFunctionCallNode
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1267
↓ 1 callersMethodAddHostComputes
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:942
↓ 1 callersMethodAddIdentityNode
tao/tao_bridge/passes/functionalize_cond.cc:810
↓ 1 callersMethodAddIfNode
tao/tao_bridge/passes/functionalize_cond.cc:821
↓ 1 callersMethodAddInputs
tao/tao_bridge/tf/lower_if_op.cc:154
↓ 1 callersMethodAddMerge
tao/tao_bridge/passes/functionalize_cond.cc:329
↓ 1 callersMethodAddOutputs
tao/tao_bridge/tf/lower_if_op.cc:172
↓ 1 callersMethodAddShapeCallCount
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:485
↓ 1 callersMethodAddShapeInferenceInfo
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1204
↓ 1 callersFunctionAllTensorTypeAnalyzed
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:139
↓ 1 callersMethodAllocateEntry
tao/tao_bridge/kernels/tao_compilation_cache.cc:70
↓ 1 callersFunctionAllocateTaoBridgeFlags
tao/tao_bridge/common.cc:28
↓ 1 callersFunctionAllocateTaoDumperFlags
tao/tao_bridge/dumper_common.cc:29
↓ 1 callersFunctionAlwaysForwardsRefInput
tao/tao_bridge/tf/xla_cluster_util.cc:91
↓ 1 callersMethodAncestorsForSuccessor
Returns the set of (subgraph, oc_cluster) pairs that should be recorded as ancestors for any successor of this node. If the node is in the outer graph
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2174
↓ 1 callersFunctionAppendIOAttr
tao_compiler/decoupling/mlir_compiler.cc:169
↓ 1 callersFunctionApplyCpuOptionsAfterTranslatingToLLVM
tao_compiler/mlir/disc/disc_compiler.cc:767
↓ 1 callersFunctionApplyCpuOptionsBeforeTranslatingToLLVM
tao_compiler/mlir/disc/disc_compiler.cc:698
↓ 1 callersFunctionAreAllParentsGuaranteedConst
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:86
↓ 1 callersFunctionArrangeOperandsInsertPointInBlock
Arrange the insert-point of `op`'s operands in the same block. Do not deal with cross-block operands.
tao_compiler/mlir/disc/transforms/disc_quantized_dot_merge.cc:38
↓ 1 callersMethodAsPair
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:1910
↓ 1 callersFunctionAsciiStrToLower
pytorch_blade/pytorch_blade/torch-mlir/lib/utils/env.cpp:19
↓ 1 callersFunctionBackUpAndSetEnv
tao/tao_bridge/dumper_common_test.cc:32
↓ 1 callersFunctionBackUpAndSetEnv
tao/tao_bridge/common_test.cc:32
↓ 1 callersFunctionBackwardDFS
tao/tao_bridge/tf/graphcycles.cc:228
↓ 1 callersFunctionBackwardDFS
Follows the edges from consumer to producer and visit all the nodes that is reachable from node `n` and have rank larger than `lower_bound`.
tao_compiler/mlir/disc/utils/cycle_detector.cc:177
↓ 1 callersFunctionBeginClusterReplayRecord
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:195
↓ 1 callersFunctionBinaryStrToSharedLibrary
tao_compiler/mlir/disc/disc_compiler.cc:888
↓ 1 callersFunctionBindInputs
tao_compiler/mlir/disc/tools/disc-replay/disc_interpreter.cc:47
↓ 1 callersFunctionBlockToONNX
pytorch_blade/pytorch_blade/compiler/jit/torch/onnx.cpp:52
↓ 1 callersMethodBuildAndReplace
tao/tao_bridge/passes/functionalize_cond.cc:772
↓ 1 callersFunctionBuildBlockGraphCycles
tao_compiler/mlir/disc/transforms/disc_quantized_dot_merge.cc:128
↓ 1 callersFunctionBuildBlockGraphCycles
tao_compiler/mlir/disc/transforms/disc_dot_merge.cc:54
↓ 1 callersMethodBuildEngine
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_onnx_parser.cpp:66
↓ 1 callersMethodBuildFunctionDef
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1122
↓ 1 callersMethodBuildFunctionDefs
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1634
↓ 1 callersFunctionBuildLoopBody
Builds a graph for the loop body.
tao/tao_bridge/passes/functionalize_while.cc:185
↓ 1 callersFunctionBuildLoopCondition
Builds a graph for the loop condition.
tao/tao_bridge/passes/functionalize_while.cc:149
↓ 1 callersMethodBuildOutputGraph
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2522
↓ 1 callersMethodCMD
tao/tao_bridge/kernels/process.cc:41
↓ 1 callersMethodCWD
tao/tao_bridge/kernels/process.cc:59
↓ 1 callersFunctionCallDiscCompiler
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/register_disc_class.cpp:70
↓ 1 callersMethodCaptureEnvVars
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:184
↓ 1 callersFunctionCastBoundaryScalarToTensor
Give: with prim::FusionGroup( %p0: Tensor, %p1: Tensor, %p2: Scalar): %1 Tensor = aten::add(%p0, %p1, %p2) return %1 Execute: CastToTensorInputs(sub_
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/disc_fuser.cpp:58
↓ 1 callersFunctionCastingScalarInputToTensor
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/disc_fuser.cpp:108
↓ 1 callersFunctionCeilOfRatio
tao/tao_bridge/tf/util.h:389
↓ 1 callersFunctionCheckSoftPlacement
tao/tao_bridge/passes/functionalize_control_flow.cc:100
↓ 1 callersFunctionCheckTarCommand
tao_compiler/mlir/disc/tools/disc-replay/record.cc:42
↓ 1 callersFunctionClassOpKind
pytorch_blade/pytorch_blade/ltc/include/torch/csrc/lazy/ts_backend/ts_node.h:100
↓ 1 callersFunctionCloneAndReplaceElementType
tao_compiler/mlir/disc/transforms/element_type_converter.cc:55
↓ 1 callersFunctionCloneConstant
tao/tao_bridge/passes/tao_clone_constants_for_better_clustering_test.cc:54
↓ 1 callersFunctionClusterInfoToProtobuf
tao/tao_bridge/tf/xla_cluster_util.cc:357
↓ 1 callersFunctionCommonFactors
tao/tao_bridge/tf/util.cc:217
↓ 1 callersMethodCompile
tao/tao_bridge/kernels/tao_compilation_cache.cc:1085
↓ 1 callersMethodCompile
tao_compiler/decoupling/mlir_compiler.cc:422
↓ 1 callersFunctionCompressTar
tao_compiler/mlir/disc/tools/disc-replay/tar_helper.h:50
↓ 1 callersFunctionComputeIncompatibleResourceOperationPairs
tao/tao_bridge/tf/resource_operation_safety_analysis.cc:259
↓ 1 callersMethodConnectSequencerToCallNode
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1076
↓ 1 callersMethodContains
tao_compiler/mlir/disc/utils/cycle_detector.h:81
↓ 1 callersFunctionConvReifyReturnTypeImpl
tao_compiler/mlir/disc/IR/hlo_disc_ops.cc:467
↓ 1 callersFunctionConvertGraphToModule
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:173
↓ 1 callersFunctionConvertInputInfo
tao_compiler/decoupling/mlir_compiler.cc:89
↓ 1 callersFunctionConvertOutputInfo
tao_compiler/decoupling/mlir_compiler.cc:134
↓ 1 callersFunctionConvertTorchScriptToMhlo
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:269
↓ 1 callersFunctionConvertTorchToMhlo
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:178
↓ 1 callersFunctionCopyIncomingControlEdges
tao/tao_bridge/passes/tao_defuse_pass.cc:50
↓ 1 callersFunctionCopyIncomingControlEdges
tao/tao_bridge/passes/tao_build_tao_op_pass.cc:122
↓ 1 callersFunctionCopyShapeInferenceNodeToGraph
Adds a copy of node_in to graph_out and adds the mapping to copied_node_images.
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1933
↓ 1 callersMethodCreate
pytorch_blade/pytorch_blade/compiler/mlir/runtime/disc_engine.cpp:94
↓ 1 callersFunctionCreateAllowAllOperationFilter
tao/tao_bridge/tf/compilability_check_util.cc:559
↓ 1 callersFunctionCreateCycleDetectionGraph
tao/tao_bridge/tf/xla_cluster_util.cc:113
↓ 1 callersFunctionCreateDiscGpuKernelToBlobPass
tao_compiler/mlir/disc/transforms/disc_gpu_kernel_to_blob.cc:439
↓ 1 callersFunctionCreateFallbackFunction
tao/tao_bridge/passes/tao_build_tao_op_pass.cc:279
↓ 1 callersFunctionCreateGraph
tao/tao_bridge/tao_util_test.cc:32
↓ 1 callersFunctionCreateGraph
tao/tao_bridge/passes/tao_defuse_pass_test.cc:27
↓ 1 callersFunctionCreateInferRuntime
tensorflow_blade/src/tensorrt/bridge/tensorrt_common.cpp:72
↓ 1 callersFunctionCreateInferRuntime
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_common.cpp:72
↓ 1 callersFunctionCreateMlirFunction
tao/tao_bridge/passes/tao_build_tao_op_pass.cc:206
↓ 1 callersFunctionCreateMultipleNextIterationInputsError
tao/tao_bridge/tf/deadness_analysis.cc:999
↓ 1 callersFunctionCreateOperationFilter
tao/tao_bridge/tf/compilability_check_util.cc:541
↓ 1 callersMethodCreatePivotNodes
tao/tao_bridge/tf/lower_if_op.cc:113
↓ 1 callersFunctionCreateResourceOpInfoMap
tao/tao_bridge/tf/resource_operation_table.cc:43
↓ 1 callersFunctionCurrentLibLocation
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/register_disc_class.cpp:31
↓ 1 callersFunctionDeCompressTar
tao_compiler/mlir/disc/tools/disc-replay/tar_helper.h:22
↓ 1 callersMethodDebugString
tao/tao_bridge/tf/deadness_analysis.cc:1617
← previousnext →1,301–1,400 of 6,088, ranked by callers