MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

↓ 2 callersFunctionfindLastWriterInBlock
tao_compiler/mlir/disc/utils/source_emitter.cc:219
↓ 2 callersFunctionfindTransposeProducer
tao_compiler/mlir/disc/transforms/disc_transpose_simplifier.cc:640
↓ 2 callersFunctionfind_trace_inputs
(func_varnames, chunk_varnames, inputs)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/torchscript_optimize.py:28
↓ 2 callersMethodforward
(self, x)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/distributed/utils.py:28
↓ 2 callersMethodforward
(self, x)
pytorch_blade/tests/test_tools.py:186
↓ 2 callersMethodforward
(self, x, y)
pytorch_blade/tests/disc/test_multi_streams.py:24
↓ 2 callersMethodforward_inplace
(self, x)
pytorch_blade/tests/tensorrt/test_support_info.py:68
↓ 2 callersMethodforward_no_inplace
(self, x)
pytorch_blade/tests/tensorrt/test_support_info.py:93
↓ 2 callersMethodfree
tao_compiler/mlir/xla/ral/context/common_context_impl_acl.cc:115
↓ 2 callersFunctionfromHex
Convert hexadecimal string \p Input to its binary representation. The return string is half the size of \p Input.
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:264
↓ 2 callersMethodfrom_qparams
(cls, qparams: QParams)
tools/torch_quant/torch_quant/observer.py:165
↓ 2 callersFunctionfuse_modules
Fuse modules for quantization Fuses only the following sequence of modules: Linear + ReLU -> LinearRelu Conv2d + ReLU -> Conv2dRe
tools/torch_quant/torch_quant/graph.py:214
↓ 2 callersMethodfusibilityAnalysis
tao_compiler/mlir/disc/transforms/fusion_utils.cc:1726
↓ 2 callersFunctionfusionTypeToString
tao_compiler/mlir/disc/transforms/fusion_utils.cc:174
↓ 2 callersFunctiongen_dummy_inputs
(device, batch_size=1, seq_len=128)
tools/torch_quant/bert_ptq_demo.py:77
↓ 2 callersFunctiongen_oss_path
(url, checksum_value, checksum_type)
scripts/python/bazel_snapshot_client.py:87
↓ 2 callersMethodgen_simple_conv_net
(self, batch: Optional[int] = 1)
tensorflow_blade/tests/tf_test_common.py:64
↓ 2 callersFunctiongenerate_copyright
(template, lang='go')
scripts/pre-commit/copyright.py:47
↓ 2 callersFunctiongetCastCompatibleMemRefType
Given two MemRefTypes `aT` and `bT`, return a MemRefType to which both can be cast. If the MemRefTypes don't have the same rank or are not strided, re
tao_compiler/mlir/disc/tools/disc-transform/TransformOps/TransformOpsExt.cc:640
↓ 2 callersFunctiongetCmdOption
pytorch_blade/tests/torchscript/shape_analysis_tool.cpp:78
↓ 2 callersFunctiongetDefaultPlacementForBackend
tao_compiler/mlir/disc/tests/mlir_test.cc:107
↓ 2 callersFunctiongetDimSizeValue
tao_compiler/mlir/disc/transforms/codegen_utils.cc:131
↓ 2 callersFunctiongetDimValues
tao_compiler/mlir/disc/tools/disc-transform/TransformOps/TransformOpsExt.cc:1317
↓ 2 callersFunctiongetDims
tao_compiler/mlir/disc/tools/disc-transform/LinalgExt/LinalgExtOps.cc:94
↓ 2 callersMethodgetDims
tao_compiler/mlir/disc/transforms/shape_utils.h:56
↓ 2 callersFunctiongetDirectConsumerOps
tao_compiler/mlir/disc/transforms/fusion_utils_dot_gpu.cc:50
↓ 2 callersFunctiongetDispatchOpSignatureEncoding
Encodes a ral_dispatch op and appends the encoding to the string buffer `out`. The format: encoding = separator.join(target_name, device, inputs_encod
tao_compiler/mlir/disc/transforms/disc_to_llvm.cc:121
↓ 2 callersMethodgetEdgePaddingLow
Returns the `edgePaddingLow` for `axis`.
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:215
↓ 2 callersFunctiongetIndicesForAccess
Return the SSA values that represent the data point accessed using a given `indexingMap` for a given point in the iteration space represented by `ivs`
tao_compiler/mlir/disc/tools/disc-transform/LinalgExt/LinalgExtOps.cc:661
↓ 2 callersMethodgetInteriorPadding
Returns the `InteriorPadding` for `axis`.
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:231
↓ 2 callersFunctiongetLinearIndex
tao_compiler/mlir/disc/transforms/revise_kernel_outlining.cc:78
↓ 2 callersFunctiongetMapOpInReduceRegion
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:377
↓ 2 callersFunctiongetMemRefValueSymbolicDims
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:1114
↓ 2 callersMethodgetNextUniqueNameOfCompiledResultProto
Returns a (per-process-level) unique name used for saving compiled result proto on disc.
tao/tao_bridge/kernels/tao_compilation_cache.cc:408
↓ 2 callersMethodgetNumDeviceConstantEmitted
Returns the number of device consts that have been emitted.
tao_compiler/mlir/xla/ral/ral_metadata.h:93
↓ 2 callersMethodgetNumHostConstantEmitted
Returns the number of host consts that have been emitted.
tao_compiler/mlir/xla/ral/ral_metadata.h:87
↓ 2 callersFunctiongetNumTransposeOps
tao_compiler/mlir/disc/transforms/disc_transpose_simplifier.cc:1021
↓ 2 callersFunctiongetOrCreateMatmulPrimitive
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.cc:684
↓ 2 callersFunctiongetOrCreateUnshapedType
pytorch_blade/pytorch_blade/compiler/jit/torch/shape_analysis.cpp:2723
↓ 2 callersFunctiongetReduceOperator
TODO: only support the reduce body in the form of one lhlo_instruction and one terminator
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.cc:145
↓ 2 callersFunctiongetReductionTileSizeOnCPU
tao_compiler/mlir/disc/transforms/codegen_utils.cc:348
↓ 2 callersFunctiongetRootMemRef
tao_compiler/mlir/disc/IR/qconv_custom_call_op.cc:170
↓ 2 callersMethodgetSortedParallelAxes
Returns the sorted parallel axes.
tao_compiler/mlir/disc/transforms/fusion_utils.h:608
↓ 2 callersMethodgetSymbolDims
tao_compiler/mlir/disc/transforms/shape_utils.h:54
↓ 2 callersMethodgetTaggedPatternStr
tao_compiler/mlir/disc/transforms/disc_transform_schedule.cc:1414
↓ 2 callersFunctiongetTypeEncoding
Encodes a mlir type and appends the encoding to the string buffer `out`.
tao_compiler/mlir/disc/transforms/disc_to_llvm.cc:71
↓ 2 callersFunctiongetVectorOffset
During unrolling from `originalShape` to `targetShape` return the offset for the slice `index`.
tao_compiler/mlir/disc/tools/disc-transform/TransformOps/TransformOpsExt.cc:1611
↓ 2 callersFunctionget_binary_version
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:85
↓ 2 callersMethodget_calib_data_for_each_group
(self)
pytorch_blade/torch_blade/quantization/prepare_data.py:192
↓ 2 callersFunctionget_fake_quant_node
(graph)
pytorch_blade/torch_blade/quantization/__init__.py:123
↓ 2 callersFunctionget_fullpath_or_create
(dir_path)
pytorch_blade/torch_blade_build.py:28
↓ 2 callersMethodget_func
(self, graph, name)
tao/tao_bridge/test/gpu/mlir/base.py:48
↓ 2 callersMethodget_groups
(self)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:25
↓ 2 callersMethodget_md5_for_packed_weight
tao_compiler/mlir/xla/ral/context/common_context_impl_acl.cc:365
↓ 2 callersFunctionget_node_name_from_input
(input_name: Optional[str])
tensorflow_blade/tf_blade/util/tf_graph_transform_util.py:588
↓ 2 callersFunctionget_node_name_parts_from_input
(input_name: str)
tensorflow_blade/tf_blade/util/tf_graph_transform_util.py:573
↓ 2 callersMethodget_or_create_packed_bias
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.h:94
↓ 2 callersMethodget_reachable
Get all nodes that can reach to the target_idx. Parameters ---------- target_idx : Set[int] The index of target n
tensorflow_blade/tf_blade/util/simple_graph.py:233
↓ 2 callersFunctionget_tf_info
(python_executable)
pytorch_blade/common_setup.py:389
↓ 2 callersFunctionget_torch_bert_large_model
(amp: bool = False)
examples/PyTorch/Inference/CUDA/BERT/main.py:103
↓ 2 callersMethodget_value
tao_compiler/mlir/xla/ral/context/custom_library/tf_topk.cu.h:55
↓ 2 callersMethodhas_cycle
(self)
pytorch_blade/torch_blade/algorithm/directed_graph.py:58
↓ 2 callersMethodhas_cycle
(self)
pytorch_blade/torch_blade/algorithm/directed_graph.py:114
↓ 2 callersMethodhas_functional_control_flow
True if the cluster has functional control flow like `If` and `While`.
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:189
↓ 2 callersMethodhas_same_shape_as
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:251
↓ 2 callersFunctionhexDigitValue
Interpret the given character \p C as a hexadecimal digit and return its value. If \p C is not a valid hex digit, -1U is returned.
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:243
↓ 2 callersFunctionhexFromNibbles
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:254
↓ 2 callersFunctioninferTypeAndPermutation
tao_compiler/mlir/disc/transforms/disc_custom_call_rewriter.cc:33
↓ 2 callersFunctioninitFusionPatternBase
tao_compiler/mlir/disc/transforms/fusion_utils.cc:542
↓ 2 callersFunctioninitSkeletonGrpsAndCloneOps
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:2705
↓ 2 callersFunctioninsertTieShapeOnRegion
tao_compiler/mlir/disc/transforms/disc_shape_optimization.cc:140
↓ 2 callersFunctioninsertTieShapeOnValue
Insert disc_shape.tie_shape for each value having RankedTensorType type. before ``` %0 = ... : tensor<?x10xf32> use(%0) ``` after ``` %0 = ... : tenso
tao_compiler/mlir/disc/transforms/disc_shape_optimization.cc:70
↓ 2 callersFunctioninsert_act_observer
(ctx: GraphModContext)
tools/torch_quant/torch_quant/graph.py:239
↓ 2 callersMethodinsert_observer
Insert the DataCollectObserver to the graph. For convenience, we group all inputs of each fusion group into a tensor list. Afte
pytorch_blade/torch_blade/quantization/prepare_data.py:122
↓ 2 callersFunctionisCompIntensFusionEnabled
tao_compiler/mlir/disc/disc_util.cc:273
↓ 2 callersFunctionisConstOneBcast
convert: %0 = mhlo.constant dense<1> : tensor<f32> %1 = bcast %0, ... : tensor<?x?xf32> %2 = mhlo.mul %1, %arg : tensor<?x?xf32> %use(%2) to: %use(%ar
tao_compiler/mlir/disc/transforms/disc_transpose_simplifier.cc:170
↓ 2 callersMethodisDynamic
tao_compiler/mlir/disc/IR/disc_shape_ops.cc:331
↓ 2 callersFunctionisExpandLikeReshape
tao_compiler/mlir/disc/transforms/fusion_utils.cc:1350
↓ 2 callersFunctionisFusibleSparseReductionOp
tao_compiler/mlir/disc/transforms/fusion_utils_sparse_op_cpu.cc:35
↓ 2 callersFunctionisGpuLmhlo
tao_compiler/mlir/disc/transforms/placement_utils.cc:33
↓ 2 callersFunctionisInFusionBlock
return true if op in fusion block
tao_compiler/mlir/disc/transforms/disc_lower_to_library_call.cc:109
↓ 2 callersMethodisInnerMostDim
Returns true if the dimension `d` is the inner most.
tao_compiler/mlir/disc/tools/disc-transform/TransformOps/TransformOpsExt.cc:1031
↓ 2 callersMethodisKInputFusion
Returns true if this fusion pattern is a kInput fusion.
tao_compiler/mlir/disc/transforms/fusion_utils.h:336
↓ 2 callersFunctionisNonBatchingTransposeTensorValue
It does not support transpose of batching dimension and other dimensions together now. TODO: it may support the transpose between batching dimensions
tao_compiler/mlir/disc/transforms/dot_rewriter.cc:48
↓ 2 callersFunctionisNonBatchingTransposeTensorValue
It does not support transpose of batching dimension and other dimensions together now. TODO: it may support the transpose between batching dimensions
tao_compiler/mlir/disc/transforms/quantized_dot_rewriter.cc:32
↓ 2 callersFunctionisOpFusible
tao_compiler/mlir/disc/transforms/fusion_utils_dot_gpu.cc:97
↓ 2 callersFunctionisRank2or3Transpose
Return true if this op is a rank-2 transpose
tao_compiler/mlir/disc/transforms/fusion_utils.cc:489
↓ 2 callersFunctionisScalarConstOp
tao_compiler/mlir/disc/transforms/fusion_utils_transform_based.cc:43
↓ 2 callersFunctionisSmallParallelOp
tao_compiler/mlir/disc/transforms/disc_cpu_map_parallel_loop.cc:46
↓ 2 callersFunctionisStdOnTensor
tao_compiler/mlir/disc/transforms/placement_utils.h:82
↓ 2 callersMethodisStitchFusion
Returns true if the fusion type is stitch fusion.
tao_compiler/mlir/disc/transforms/fusion_utils.h:342
↓ 2 callersFunctionisSupportedBcast
tao_compiler/mlir/disc/transforms/fusion_utils_transform_based.cc:55
↓ 2 callersMethodisSupportedOp
tao_compiler/mlir/disc/utils/source_emitter.cc:633
↓ 2 callersFunctionisTrueConstant
pytorch_blade/pytorch_blade/compiler/jit/torch/const_loop_unroll.cpp:22
↓ 2 callersFunctionis_aarch64
()
pytorch_blade/common_setup.py:353
↓ 2 callersMethodis_default
tao/tao_bridge/executable.h:60
↓ 2 callersFunctionis_fake_quant_op
(inp_node_kind)
pytorch_blade/torch_blade/quantization/__init__.py:101
↓ 2 callersFunctionis_gpu_tensor_type
pytorch_blade/pytorch_blade/compiler/jit/tool_funcs.cpp:97
↓ 2 callersMethodis_quantizable
(self, module_name: str)
tools/torch_quant/torch_quant/graph.py:124
↓ 2 callersFunctionjit_add_observer
(graph, jit_obs, value, method_name='record')
pytorch_blade/torch_blade/tools/shape_inference.py:45
← previousnext →1,101–1,200 of 6,088, ranked by callers