MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / mpu

Method mpu

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

Source from the content-addressed store, hash-verified

564 return self._topo
565
566 def mpu(self):
567 return self._grid
568
569 def num_pipeline_stages(self):
570 return self._topo.get_dim('pipe')

Callers 1

initializeFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected