github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-05 06:58:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
afb239bbf8
ColossalAI/tests/test_device
History
Frank Lee 80eba05b0a
[test] refactor tests with spawn () 
* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-04-06 14:51:35 +08:00
..
test_alpha_beta.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
test_device_mesh.py [device] add DeviceMesh class to support logical device layout () 2022-08-02 19:23:48 +08:00
test_extract_alpha_beta.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
test_init_logical_pg.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
test_search_logical_device_mesh.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00