Function parallel_model

examples/distributed/parallel_opt.py:9–247 · 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)

start_workerFunction · 0.90

ReplicateClass · 0.90

find_successorFunction · 0.85

shard_binaryFunction · 0.85

shard_nodeFunction · 0.85

is_partialMethod · 0.80

shard_gemmFunction · 0.70

no test coverage detected