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
↓ 3 callers
Method
register_pipeline
(cls, name, func)
pytorch_blade/torch_blade/config.py:28
↓ 3 callers
Method
releaseHostConstant
tao_compiler/mlir/xla/ral/ral_metadata.cc:101
↓ 3 callers
Method
replace_root
tao_compiler/mlir/xla/ral/context/custom_library/tf_topk.cu.h:181
↓ 3 callers
Function
restore_env_var
(name, saved)
scripts/python/tao_common.py:128
↓ 3 callers
Function
running_on_ci
Return true if the building job is running on CI host.
scripts/python/common_setup.py:198
↓ 3 callers
Function
running_on_ci
Return true if the building job is running on CI host.
pytorch_blade/common_setup.py:198
↓ 3 callers
Method
setSpecificLoader
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.h:126
↓ 3 callers
Method
setUpWithMLIR
(testcase='')
tao/tao_bridge/test/gpu/mlir/base.py:27
↓ 3 callers
Method
set_allocation
tao/tao_bridge/executable.h:125
↓ 3 callers
Method
set_deleter
tao_compiler/mlir/xla/ral/context/base/base_context.h:54
↓ 3 callers
Function
set_extra_inputs
(c_module, extra_inputs, extra_dynamic_shapes)
pytorch_blade/torch_blade/tensorrt/dynamic_shapes.py:131
↓ 3 callers
Method
set_scale
Set new scale into param
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:895
↓ 3 callers
Method
set_zero_point
Set new scale into param
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:911
↓ 3 callers
Function
str2format
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.cc:162
↓ 3 callers
Function
symlink_disc_files
(args)
scripts/python/common_setup.py:606
↓ 3 callers
Function
tao_bridge_bazel_config
(args)
scripts/python/tao_build.py:451
↓ 3 callers
Function
tao_bridge_dir
(root=None)
scripts/python/common_setup.py:537
↓ 3 callers
Function
toPlacementType
tao_compiler/mlir/disc/transforms/mhlo_placer.cc:38
↓ 3 callers
Method
toString
pytorch_blade/pytorch_blade/ltc/disc_backend/backend_impl.cpp:56
↓ 3 callers
Method
to_functiondef
Generate subgraph FunctionDef from this segment. Parameters ---------- seg_id : int = 0 The id of this segment/su
tensorflow_blade/tf_blade/util/simple_graph.py:459
↓ 3 callers
Method
topological_sort
Sort given SimpleGraph in topological order. Parameters ---------- reverse : bool = False Set True to list op fro
tensorflow_blade/tf_blade/util/simple_graph.py:167
↓ 3 callers
Function
trace_model
(model: PreTrainedModel, info: ModelInfo, amp: bool = False)
examples/PyTorch/Inference/hf_transformers/blade_adapter.py:256
↓ 3 callers
Method
update
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:155
↓ 3 callers
Function
updateFunctionType
Updates the function types according to the types of entry block arguments and the types of operands of the return op of the func op. This function su
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:800
↓ 3 callers
Function
warpFilterCompress
tao_compiler/mlir/xla/ral/context/custom_library/warp.cu.h:144
↓ 3 callers
Function
warpReduceMaxWithPos
tao_compiler/mlir/xla/ral/context/custom_library/warp.cu.h:62
↓ 3 callers
Function
wrap_model_inputs
wrap forward func to ignore some inputs
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/optimization/utils.py:93
↓ 3 callers
Method
write_version_file
(self, version_path)
pytorch_blade/torch_blade_build.py:112
↓ 2 callers
Function
AddDtypeToKernalDefConstraint
tao/tao_bridge/tf/xla_op_registry.cc:628
↓ 2 callers
Method
AddExecutable
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:670
↓ 2 callers
Function
AddMissingFunctionByName
Copy the FunctionDef of given function from lookup_library to library, if it can be found in lookup_library but is missing from library.
tao/tao_bridge/passes/functionalize_while.cc:237
↓ 2 callers
Function
AddMissingFunctionDef
Iterate over all functions that the given fdef refers to. Copy the missing FunctionDefs from lookup_library to library.
tao/tao_bridge/passes/functionalize_while.cc:248
↓ 2 callers
Function
Align
std::align is not supported, so this method mimic its behavior.
tensorflow_blade/src/util/tf_allocator_util.cc:87
↓ 2 callers
Method
AllNodesInPostOrder
tao/tao_bridge/tf/graphcycles.cc:432
↓ 2 callers
Method
AllocateDeviceMemory
tensorflow_blade/src/tensorrt/bridge/tensorrt_tf_allocator.cc:88
↓ 2 callers
Method
AncestorStateToString
tao/tao_bridge/passes/functionalize_cond.cc:235
↓ 2 callers
Function
AppendMarkForCompilationPassFlagsInternal
tao/tao_bridge/tf/flags.cc:76
↓ 2 callers
Function
ArrangeOperandsInsertPointInBlock
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_dot_merge.cc:42
↓ 2 callers
Function
BroadcastTensorRanks
pytorch_blade/pytorch_blade/torch-mlir/lib/Conversion/TorchToMhlo/DiscTorchToMhlo.cpp:56
↓ 2 callers
Function
BuildArgNode
tao/tao_bridge/passes/functionalize_while.cc:138
↓ 2 callers
Method
BuildEngine
tensorflow_blade/src/tensorrt/bridge/tensorrt_onnx_parser.cpp:72
↓ 2 callers
Function
CallStatusReturningFunction
tao/tao_bridge/errors_test.cc:71
↓ 2 callers
Method
CanContractEdge
tao_compiler/mlir/disc/utils/cycle_detector.cc:346
↓ 2 callers
Function
CandidateCudaRoots
tao_compiler/mlir/disc/transforms/legalize_to_cubin.cc:91
↓ 2 callers
Function
CeilOfRatio
tao_compiler/mlir/xla/ral/context/custom_library/transpose_gpu.cu.cc:21
↓ 2 callers
Function
CheckEdges
tao/tao_bridge/tf/graphcycles_test.cc:158
↓ 2 callers
Function
CheckNodeNotInCycle
Check that the graph has no cycle containing the given node.
tao/tao_bridge/passes/functionalize_control_flow_util.cc:58
↓ 2 callers
Function
CheckTransitiveClosure
tao/tao_bridge/tf/graphcycles_test.cc:135
↓ 2 callers
Method
Clear
tao/tao_bridge/tf/ordered_set.h:68
↓ 2 callers
Function
ClearVisitedBits
tao/tao_bridge/tf/graphcycles.cc:291
↓ 2 callers
Function
ClearVisitedBits
Clears bookkeeping fileds used during the last DFS process.
tao_compiler/mlir/disc/utils/cycle_detector.cc:248
↓ 2 callers
Function
CommonVerifyForQuantizedComputeIntensiveOp
tao_compiler/mlir/disc/IR/hlo_disc_ops.cc:352
↓ 2 callers
Method
Compile
tao_compiler/mlir/disc/tools/disc-replay/disc_interpreter.cc:26
↓ 2 callers
Function
CompileToDiscExecutable
pytorch_blade/pytorch_blade/ltc/disc_compiler/disc_compiler.cpp:121
↓ 2 callers
Function
CondConstInputIndices
tao/tao_bridge/tf/const_analysis.cc:62
↓ 2 callers
Function
ConvertResults
Converts the results of the operation `op` to memref types and append them to the `results` vector.
tao_compiler/mlir/disc/transforms/disc_hlo_legalize_to_lhlo.cc:105
↓ 2 callers
Function
ConvertTF2MlirHlo
tao_compiler/mlir/disc/disc_compiler.cc:941
↓ 2 callers
Function
CopySubgraph
Copies a subgraph from `graph` to `output` by performing a reverse DFS starting at nodes in vector `stack`. `node_map` is a vector indexed by source n
tao/tao_bridge/passes/functionalize_while.cc:100
↓ 2 callers
Function
CustomFuseGraph
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/graph_fuser.cpp:1302
↓ 2 callers
Function
DataTypeToMlirType
tao_compiler/decoupling/mlir_compiler.cc:151
↓ 2 callers
Function
DescribeCycle
Returns a string describing how an edge from src to dst would create a cycle.
tao/tao_bridge/tf/xla_cluster_util.cc:53
↓ 2 callers
Function
DeviceTypeToStr
tao_compiler/mlir/disc/tests/mlir_feature_test.cc:46
↓ 2 callers
Function
DoAsserts
tao/tao_bridge/dumper_common_test.cc:47
↓ 2 callers
Function
DoAsserts
tao/tao_bridge/common_test.cc:47
↓ 2 callers
Function
DumpGraphDefToFile
tao/tao_bridge/tf/dump_graph.cc:91
↓ 2 callers
Function
DumpLLVMModule
tao_compiler/mlir/disc/disc_compiler.cc:84
↓ 2 callers
Method
EmitOp
tao_compiler/mlir/disc/utils/source_emitter.cc:587
↓ 2 callers
Function
FindFirstNotOf
Like s.find_first_not_of(x, pos), but return s.size() when find_first_not_of() would return string::npos. This avoids if-statements elsewhere.
tao/tao_bridge/tf/parse_flags_from_env.cc:88
↓ 2 callers
Function
FindNodesToDecluster
tao/tao_bridge/passes/tao_partially_decluster_pass.cc:41
↓ 2 callers
Function
FindUniqueBackedge
tao/tao_bridge/tf/deadness_analysis.cc:1012
↓ 2 callers
Method
ForEach
tao/tao_bridge/tf/device_util.h:74
↓ 2 callers
Function
ForwardDFS
tao/tao_bridge/tf/graphcycles.cc:200
↓ 2 callers
Function
ForwardDFS
Follows the edges from producer to consumer and searchs if the node having rank `n` can reach the node having rank `upper_bound` using a DFS search. W
tao_compiler/mlir/disc/utils/cycle_detector.cc:147
↓ 2 callers
Function
FusionOutputs
pytorch_blade/pytorch_blade/ltc/disc_compiler/replay.cpp:43
↓ 2 callers
Function
GenerateMlirModuleString
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:54
↓ 2 callers
Function
GenerateStrSeq
tao_compiler/mlir/disc/tests/mlir_feature_test.cc:36
↓ 2 callers
Method
GetBackendDevice
pytorch_blade/pytorch_blade/ltc/disc_backend/backend_impl.cpp:268
↓ 2 callers
Function
GetBackendName
pytorch_blade/pytorch_blade/compiler/tensorrt/tensorrt_engine.cpp:68
↓ 2 callers
Function
GetCandidateRowReduceOp
tao_compiler/mlir/disc/transforms/disc_specialize_fusion_with_speculation.cc:187
↓ 2 callers
Function
GetCheckPointId
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:1656
↓ 2 callers
Function
GetCompileTimeConstInputs
tao/tao_bridge/tf/const_analysis.cc:100
↓ 2 callers
Function
GetCompilerFactories
tao_compiler/decoupling/compiler_base.cc:22
↓ 2 callers
Function
GetConvMetadata
tao_compiler/mlir/disc/transforms/disc_lower_to_library_call.cc:274
↓ 2 callers
Function
GetDebugNames
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:89
↓ 2 callers
Function
GetDiscCpuMathKernelMode
tao_compiler/mlir/xla/ral/context/common_context_impl_mkldnn.cc:125
↓ 2 callers
Method
GetLeaderValueWithSameShapeInFusion
tao_compiler/mlir/disc/transforms/shape_utils.cc:1825
↓ 2 callers
Function
GetMicroBatchId
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:1642
↓ 2 callers
Method
GetOutsideCompilationSubgraphNames
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:1435
↓ 2 callers
Function
GetPointerToFuel
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2449
↓ 2 callers
Method
GetRecvAtHostNode
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:731
↓ 2 callers
Function
GetRootFrame
If the node is inside some frames, get the name of the outermost non-empty frame. Otherwise, get an empty frame name.
tao/tao_bridge/tf/deadness_analysis.cc:1093
↓ 2 callers
Function
GetShapeFromConstUniqueName
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:217
↓ 2 callers
Method
GetShapeNum
tao/tao_bridge/kernels/tao_compilation_info_collector.h:110
↓ 2 callers
Function
GetSingletonState
tao/tao_bridge/tf/xla_config_registry.cc:32
↓ 2 callers
Method
GetStatus
Returns the status. Used by MakeErrorStreamWithOutput.
tao/tao_bridge/errors.h:153
↓ 2 callers
Function
GetTensorAddress
tensorflow_blade/src/custom_ops/trt_engine_op/trt_engine_op.cc:57
↓ 2 callers
Function
GetXlaAutoClusteringSummary
tao/tao_bridge/tf/xla_cluster_util.cc:366
↓ 2 callers
Function
GpuLaunchKernel
tao_compiler/mlir/xla/ral/context/custom_library/gpu_helper.h:49
↓ 2 callers
Method
HasOverflowed
tao/tao_bridge/tf/deadness_analysis.cc:603
↓ 2 callers
Function
HasResourceInputOrOutput
tao/tao_bridge/tf/xla_cluster_util.cc:215
← previous
next →
801–900 of 6,088, ranked by callers