Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/alibaba/BladeDISC
/ functions
Functions
6,088 in github.com/alibaba/BladeDISC
⨍
Functions
6,088
◇
Types & classes
1,279
↳
Endpoints
6
↓ 1 callers
Method
isFullySlicedAxis
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:927
↓ 1 callers
Method
isFusableDefault
Default fusability check - used when the user doesn't pass in a callback.
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/graph_fuser.cpp:209
↓ 1 callers
Method
isFusible
tao_compiler/mlir/disc/transforms/fusion_utils_sparse_op_cpu.cc:42
↓ 1 callers
Method
isInteriorPaddingUnknown
Returns true if the `interiorPadding` of the `axis` is unknown.
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:226
↓ 1 callers
Function
isMutableTypeImpl
pytorch_blade/pytorch_blade/compiler/jit/torch/alias_analysis.cpp:64
↓ 1 callers
Method
isNotPaddedAxis
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:1035
↓ 1 callers
Function
isOpTriviallyDeadDisc
pytorch_blade/pytorch_blade/torch-mlir/lib/Dialect/TorchConversion/Transforms/ApplyDiscPdlPatterns.cpp:100
↓ 1 callers
Function
isQint32ConstantZeroTensor
tao_compiler/mlir/disc/transforms/lower_tf.cc:758
↓ 1 callers
Method
isShapeValueEqual
tao_compiler/mlir/disc/transforms/shape_utils.cc:849
↓ 1 callers
Function
isSimpleMap
What is a simple mappable operator? It: - Has a single tensor output - Output and all tensor inputs have the same shape - Output and all tensor input
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/graph_fuser.cpp:46
↓ 1 callers
Function
isSingleElementH2DOp
tao_compiler/mlir/disc/transforms/fusion_utils.cc:1445
↓ 1 callers
Function
isSinkingBeneficiary
Identifies operations that are beneficial to sink into kernels. These operations may not have side-effects, as otherwise sinking (and hence duplicatin
tao_compiler/mlir/disc/transforms/disc_outline_cpu_kernel.cc:48
↓ 1 callers
Function
isSmallCpuKernel
tao_compiler/mlir/disc/transforms/disc_cpu_map_parallel_loop.cc:81
↓ 1 callers
Function
isSmallForOp
tao_compiler/mlir/disc/transforms/disc_cpu_map_parallel_loop.cc:65
↓ 1 callers
Method
isStartIndexUnknown
Returns true if the `startIndices` of the `axis` is unknown.
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:134
↓ 1 callers
Function
isStitchEnabled
tao_compiler/mlir/disc/disc_util.cc:263
↓ 1 callers
Method
isStrideUnknown
Returns true if the `stride` of the `axis` is unknown.
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:155
↓ 1 callers
Method
isSymbolicDimProductEqual
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:300
↓ 1 callers
Function
isTensorDialect
Check Op if it is a tensor Op.
tao_compiler/mlir/disc/transforms/placement_utils.h:92
↓ 1 callers
Method
isUnfold
tao_compiler/mlir/disc/transforms/shape_utils.h:110
↓ 1 callers
Method
isWrittenBack
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.h:122
↓ 1 callers
Method
is_4c_blocked
workaround for issue intel/mkl-dnn#588
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:195
↓ 1 callers
Function
is_aarch64
()
scripts/python/common_setup.py:353
↓ 1 callers
Function
is_available
()
pytorch_blade/torch_blade/mlir/__init__.py:36
↓ 1 callers
Function
is_available
()
pytorch_blade/torch_blade/neural_engine/__init__.py:25
↓ 1 callers
Function
is_concrete_shape_tensor_type
pytorch_blade/pytorch_blade/compiler/jit/tool_funcs.cpp:88
↓ 1 callers
Method
is_default
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:142
↓ 1 callers
Method
is_entry_computation_parameter
tao/tao_bridge/executable.h:92
↓ 1 callers
Method
is_grouped
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:424
↓ 1 callers
Method
is_hwio
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:184
↓ 1 callers
Method
is_maybe_live_out
tao/tao_bridge/executable.h:97
↓ 1 callers
Function
is_per_channel
(qscheme: torch.qscheme)
tools/torch_quant/torch_quant/observer.py:94
↓ 1 callers
Function
is_per_tensor
(qscheme: torch.qscheme)
tools/torch_quant/torch_quant/observer.py:90
↓ 1 callers
Method
is_plain
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:138
↓ 1 callers
Method
is_resource
tao/tao_bridge/executable.h:59
↓ 1 callers
Function
is_symmetric
(qscheme: torch.qscheme)
tools/torch_quant/torch_quant/observer.py:98
↓ 1 callers
Method
is_temp_buffer
tao/tao_bridge/executable.h:112
↓ 1 callers
Method
is_thread_local
tao/tao_bridge/executable.h:116
↓ 1 callers
Method
is_tuple
tao/tao_bridge/executable.h:107
↓ 1 callers
Function
is_x86
()
scripts/python/common_setup.py:348
↓ 1 callers
Function
is_x86
()
pytorch_blade/common_setup.py:348
↓ 1 callers
Method
is_zero
returns true if memory descriptor is zero */
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:127
↓ 1 callers
Function
iskSparseReductionOutputFusible
tao_compiler/mlir/disc/transforms/fusion_utils_sparse_op_cpu.cc:30
↓ 1 callers
Function
iskWhereInputFusible
tao_compiler/mlir/disc/transforms/fusion_utils_sparse_op_cpu.cc:25
↓ 1 callers
Function
json_obj_add
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:44
↓ 1 callers
Function
layout_match
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:1570
↓ 1 callers
Function
lecun_normal_init_
(weights)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:60
↓ 1 callers
Function
link_files
(args)
tensorflow_blade/build.py:113
↓ 1 callers
Function
list_files_with_extensions
(dir, extensions)
examples/PyTorch/Inference/CUDA/AlphaFold/openfold_inference.py:59
↓ 1 callers
Function
loadArguments
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:64
↓ 1 callers
Function
loadDependentDialects
tao_compiler/mlir/disc/tools/disc-source-emitter/disc-cuda-emitter.cc:25
↓ 1 callers
Function
loadDependentDialects
tao_compiler/mlir/disc/tools/disc-pdll/disc-pdll.cc:39
↓ 1 callers
Function
loadDependentDialects
pytorch_blade/tests/torch-disc-pdll/torch-disc-pdll.cpp:29
↓ 1 callers
Function
loadGlobalString
Loads a global op at the current insertion point and returns the loaded value.
tao_compiler/mlir/disc/transforms/disc_to_llvm.cc:158
↓ 1 callers
Function
loadGraph
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:56
↓ 1 callers
Function
load_feature_for_one_target
( config, data_folder, seed=0, is_multimer=False, use_uniprot=False )
examples/PyTorch/Inference/CUDA/AlphaFold/unifold_inference.py:48
↓ 1 callers
Function
load_frozen_graph
(model_file : str)
examples/TensorFlow/Inference/X86/BERT/main.py:21
↓ 1 callers
Function
load_frozen_graph
(model_file : str)
examples/TensorFlow/Inference/AArch64/BERT/main.py:21
↓ 1 callers
Function
load_frozen_graph
(model_file: str)
examples/TensorFlow/Inference/CUDA/BERT/main.py:23
↓ 1 callers
Function
load_model
()
examples/PyTorch/Inference/CPU/albert/main.py:151
↓ 1 callers
Function
load_model
(params)
pytorch_blade/benchmark/torch-tensorrt/perf_run.py:369
↓ 1 callers
Function
logger_level_context
(level)
pytorch_blade/torch_blade/logging.py:24
↓ 1 callers
Function
loopUnrollByFactorAndTryInterleave
This function unrolls the given for-loop op and interleaves the instructions
tao_compiler/mlir/disc/transforms/codegen_utils.cc:524
↓ 1 callers
Function
lowerFakeQuantToQuantAndDequant
tao_compiler/mlir/disc/disc_util.cc:243
↓ 1 callers
Function
lowerWithScheduleLoopV2
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:332
↓ 1 callers
Function
lowerWithScheduleSparseFillEmptyRowsOpCPU
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:4375
↓ 1 callers
Function
lowerWithScheduleSparseReshapeOpCPU
SparseReshape implementation: Loop over input for (i = 0; i < input_indices.shape(0), ++i) { // multidim index to linear index linear_index = ...; /
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:4201
↓ 1 callers
Function
lowerWithScheduleSparseSegmentReductionOpCPU
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:4631
↓ 1 callers
Function
lowerWithScheduleStitch
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:2883
↓ 1 callers
Function
lowerWithScheduleStitchV2
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:3564
↓ 1 callers
Function
lowerWithSchedulekInputCPU
Emitter for non-row-reduction kInput fusion pattern. Take a column reduction `memref<100x1100xf32> -> memref<1100xf32>` as an example: for (int iOutt
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:5294
↓ 1 callers
Function
main
()
tao_compiler/mlir/xla/ral/test/bace.py:25
↓ 1 callers
Function
main
()
tao_compiler/mlir/xla/ral/test/codgen_kernels.py:25
↓ 1 callers
Function
main
tao_compiler/mlir/disc/disc_compiler_main.cc:241
↓ 1 callers
Function
main
()
scripts/python/tao_build.py:858
↓ 1 callers
Function
main
()
scripts/python/bazel_snapshot_client.py:144
↓ 1 callers
Function
main
(argv=None)
scripts/pre-commit/copyright.py:88
↓ 1 callers
Function
main
()
examples/PyTorch/Train/Lazy/Bert/bert_train_ltc.py:227
↓ 1 callers
Function
main
()
examples/PyTorch/Train/Dynamo/Bert/test_bert.py:255
↓ 1 callers
Function
main
(args)
examples/PyTorch/Inference/CUDA/AlphaFold/openfold_inference.py:62
↓ 1 callers
Function
main
(args)
examples/PyTorch/Inference/CUDA/AlphaFold/unifold_inference.py:75
↓ 1 callers
Function
main
()
tensorflow_blade/build.py:385
↓ 1 callers
Function
makeDynamicShapeGEMMWithBiasParamsKey
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.h:235
↓ 1 callers
Function
makeGEMMWithBiasParamsKey
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.h:200
↓ 1 callers
Function
makeGemmTuningCacheKey
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:92
↓ 1 callers
Function
makeNewDeviceStrategy
tao_compiler/mlir/disc/transforms/fusion_utils.cc:1561
↓ 1 callers
Function
makeSortDescriptor
tao_compiler/mlir/xla/ral/context/dynamic_sort_impl.h:26
↓ 1 callers
Function
makeSymbolicDimOpRefArrayAttr
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:779
↓ 1 callers
Function
make_filename
This function is copied from c10/util/tempfile.h, so it follows to these temperary directory env variables, too.
pytorch_blade/pytorch_blade/common_utils/tempfs.cpp:28
↓ 1 callers
Function
make_package
(root, args)
scripts/python/tao_build.py:632
↓ 1 callers
Method
malloc
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/allocators.hpp:13
↓ 1 callers
Function
mapResultToInitOperandOfDestinationStyleOp
tao_compiler/mlir/disc/tools/disc-transform/TransformOps/TransformOpsExt.cc:271
↓ 1 callers
Method
markAsFullySlicedAxis
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:933
↓ 1 callers
Function
matchAndRwriterQuantizedConv2DWithBiasAndRequantizeOp
tao_compiler/mlir/disc/transforms/lower_tf.cc:1033
↓ 1 callers
Function
materializeShapeComputation
materialize shape computation IR by resolving tensor.dim + op having interface InferShapedTypeOpInterface: before ``` %arg0 = ... : tensor<?x10xf32> %
tao_compiler/mlir/disc/transforms/disc_shape_optimization.cc:235
↓ 1 callers
Function
mayConvertToIntegerType
tao_compiler/mlir/disc/transforms/codegen_utils.cc:148
↓ 1 callers
Function
maybeEmitInitLoops
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:870
↓ 1 callers
Method
merge
Merges `other` into this cluster, and clears `other`.
tao_compiler/mlir/disc/transforms/disc_quantized_dot_merge.cc:66
↓ 1 callers
Method
merge
Merges `other` into this cluster, and clears `other`.
tao_compiler/mlir/disc/transforms/disc_dot_merge.cc:84
↓ 1 callers
Function
mergeAssignment
tao_compiler/mlir/disc/transforms/disc_assign_memory_space.cc:73
← previous
next →
2,301–2,400 of 6,088, ranked by callers