Files
ColossalAI/colossalai/device/__init__.py
YuliangLiu0306 0442f940f0 [device] add DeviceMesh class to support logical device layout (#1394)
* [device] add DeviceMesh class to support logical device layout

* polish code

* add doc string
2022-08-02 19:23:48 +08:00

0 lines
0 B
Python