MCPcopy Create free account
hub / github.com/InfiniTensor/InfiniTensor / shard_gemm

Function shard_gemm

examples/distributed/parallel_opt.py:46–59  ·  view source on GitHub ↗
(node: NodeProto, groups: int = 1)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

parallel_modelFunction · 0.70

Calls 4

ShardClass · 0.90
_PartialClass · 0.90
is_replicateMethod · 0.80
shard_tensorFunction · 0.70

Tested by

no test coverage detected