github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-17 07:58:15 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
44fd3c83abb7fd47cce61ff18901441c6432f877
ColossalAI/colossalai/device
History
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
..
__init__.py
[device] add DeviceMesh class to support logical device layout (#1394)
2022-08-02 19:23:48 +08:00
device_mesh.py
[device] add DeviceMesh class to support logical device layout (#1394)
2022-08-02 19:23:48 +08:00