Functions — github.com/InfiniTensor/InfiniTensor

CodeHub MCPcopy Create free account

hub / github.com/InfiniTensor/InfiniTensor / functions

Functions2,429 in github.com/InfiniTensor/InfiniTensor

⨍ Functions 2,429◇ Types & classes 632

test/kernels/cuda/test_cuda_G2BMM.cc:13

test/kernels/cuda/test_cuda_softmax.cc:67

test/kernels/cuda/test_cuda_all_gather.cc:38

test/kernels/cuda/test_cuda_broadcast.cc:41

test/kernels/cuda/test_cuda_conv_fp16.cc:50

test/kernels/cuda/test_cuda_resize.cc:9

test/kernels/cuda/test_cuda_all_reduce.cc:38

Setup communication between 2 threads, each controlling 1 MLU. Do AllReduce Sum on {1.0, 4.0}. Results should be {5.0, 5.0}.

test/bang/test_cncl_comm.cc:37

Setup communication between 2 threads, each controlling 1 GPU. Do AllReduce Sum on {1.0, 4.0}. Results should be {5.0, 5.0}.

test/cuda/test_nccl_comm.cc:37

test/cuda/test_cudagraph.cc:11

test/nnet/test_memboundOp.cc:32

test/nnet/test_OpSearch.cc:50

MethodTensorBaseObj

src/core/tensor_base.cc:6

MethodTensorNode

src/nnet/expr.cc:29

MethodTensorObj

src/core/tensor.cc:11

MethodTransposeObj

src/operators/transpose.cc:4

include/core/object.h:14

src/operators/unary.cc:4

MethodUnsqueezeObj

src/operators/unsqueeze.cc:5

include/nnet/expr.h:130

src/operators/where.cc:6

MethodWorkspaceObj

include/core/workspace.h:13

MethodXcclCommunicatorObj

include/kunlun/xccl_communicator.h:25

(self, other: object)

examples/distributed/placement.py:22

(self, other: object)

examples/distributed/placement.py:39

(self, other: object)

examples/distributed/placement.py:55

( self, model: ModelProto, runtime, use_naive_allocator: bool = False,

pyinfinitensor/src/pyinfinitensor/onnx.py:41

examples/distributed/placement.py:36

(self, reduce_op: str = "sum")

examples/distributed/placement.py:52

Function__ref_less

include/nnet/ref.h:27

machine readable representation of the Replicate placement

examples/distributed/placement.py:27

machine readable representation of the Shard placement

examples/distributed/placement.py:44

machine readable representation of the Partial placement

examples/distributed/placement.py:60

Function_parse_data_fp16

(tensor: TensorProto)

pyinfinitensor/src/pyinfinitensor/onnx.py:1543

MethodaCosCompute

src/kernels/cpu/unary.cc:56

MethodaCoshCompute

src/kernels/cpu/unary.cc:58

MethodaSinCompute

src/kernels/cpu/unary.cc:62

MethodaSinhCompute

src/kernels/cpu/unary.cc:64

MethodaTanCompute

src/kernels/cpu/unary.cc:67

MethodaTanhCompute

src/kernels/cpu/unary.cc:69

MethodabsCompute

src/kernels/cpu/unary.cc:30

MethodaddCompute

src/kernels/cpu/element_wise.cc:7

MethodaddConnection

add op as a target

src/core/graph.cc:331

include/core/graph.h:82

MethodaddOpWithOutputs

include/core/graph.h:92

MethodaddOperatorAndConnect

src/core/graph.cc:45

FunctionallGather

test/kernels/kunlun/test_kunlun_allgather.cc:14

FunctionallGather

test/kernels/ascend/test_ascend_all_gather.cc:14

FunctionallGather

test/kernels/cuda/test_cuda_all_gather.cc:15

FunctionallReduce

test/kernels/kunlun/test_kunlun_allreduce.cc:15

FunctionallReduce

test/kernels/bang/test_bang_all_reduce.cc:17

FunctionallReduce

test/kernels/ascend/test_ascend_all_reduce.cc:15

FunctionallReduce

test/kernels/cuda/test_cuda_all_reduce.cc:16

FunctionallReduceSum

test/cuda/test_nccl_comm.cc:10

include/kunlun/kunlun_runtime.h:42

include/core/runtime.h:129

include/bang/bang_runtime.h:45

include/intelcpu/mkl_runtime.h:24

include/ascend/ascend_runtime.h:43

include/cuda/cuda_runtime.h:56

include/nnet/dbg.h:760

MethodappendCanddiate

src/nnet/derivator.cc:405

include/nnet/ref.h:18

include/core/ref.h:23

Functionascend_runtime

src/ffi/ffi_infinitensor.cc:185

Functionbang_runtime

src/ffi/ffi_infinitensor.cc:175

Functionbatch_norm_attrs_of

src/ffi/ffi_infinitensor.cc:223

Functionbfp16_to_fp32

src/utils/data_convert.cc:37

include/nnet/dbg.h:144

Functionbroadcast

test/kernels/kunlun/test_kunlun_broadcast.cc:15

Functionbroadcast

test/kernels/cuda/test_cuda_broadcast.cc:16

MethodbuildDLTOuterRangeOp

src/nnet/replace_kit.cc:61

MethodbuildDLTSingleRangeOp

src/nnet/Pass/Rule7DLT.cc:60

MethodbuildExpr

include/nnet/iterator_table.h:191

MethodbuildExprTree

src/nnet/Visitor/Serializer.cc:139

MethodbuildGuidedDLTSource

src/nnet/Pass/Rule8GuidedDLT.cc:305

MethodbuildMetaGraphWithGraph

Build metagraph with a graph, each operator is a node.

src/core/search_engine.cc:115

MethodbuildMetaGraphWithPlan

Build a metagraph with graph and a plan, a plan is which ops should be a node.

src/core/search_engine.cc:160

MethodbuildRoutine

src/nnet/Visitor/Serializer.cc:191

MethodbuildSubscirptForLoopVarReplace

src/nnet/replace_kit.cc:37

src/operators/membound.cc:79

MethodcalcPadding

src/nnet/iterator_table.cc:378

Functioncast_to_of

src/ffi/ffi_infinitensor.cc:327

MethodcheckDerivationEquivalence

src/nnet/derivator.cc:518

MethodcheckElementsHaveOnlyOneAccessIteratorSet

src/nnet/Pass/Rule8GuidedDLT.cc:286

MethodcheckEmpty

src/nnet/Visitor/MergeMemboundMutator.cc:59

MethodcheckMatchValid

src/core/graph_match.cc:274

src/operators/membound.cc:83

src/kernels/cuda/membound_tvm_extract_source.cc:73

MethodcheckOverlapsWithPreviousMatch

src/core/graph_match.cc:264

MethodcheckReplacement

inputs and outputs must be appointed.

src/core/graph_match.cc:440

MethodcheckValid

src/core/operator.cc:59

pyinfinitensor/src/pyinfinitensor/onnx.py:845

Functionclip_attrs_of

src/ffi/ffi_infinitensor.cc:240

include/nnet/Visitor/CloneMutator.h:13

(self, tensor: backend.Tensor)

pyinfinitensor/src/pyinfinitensor/onnx.py:1484

include/core/graph_handler.h:142

Functioncompute

python/cpp_plugin/gen_ansor_so.py:50

src/kernels/kunlun/softmax.cc:7

src/kernels/kunlun/batch_norm.cc:7

← previous next →1,001–1,100 of 2,429, ranked by callers