github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-21 19:43:11 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
d84e7479750f820040ca53ca8bbf4589ae6f645c
ColossalAI/tests/test_device/test_device_mesh.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

785 B
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink