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

Function parallel_model

examples/distributed/parallel.py:30–103  ·  view source on GitHub ↗
(model: ModelProto, tp_world_size: int = 1, tp_rank: int = 0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls 1

shard_gemmFunction · 0.70

Tested by

no test coverage detected