MCPcopy
hub / github.com/deepspeedai/DeepSpeed / mpu

Method mpu

deepspeed/runtime/pipe/module.py:560–561  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

558 return self._topo
559
560 def mpu(self):
561 return self._grid
562
563 def num_pipeline_stages(self):
564 return self._topo.get_dim('pipe')

Callers 1

initializeFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected