ColossalAI/tests/test_device
Frank Lee 80eba05b0a
[test] refactor tests with spawn (#3452)
* [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 (#3452) 2023-04-06 14:51:35 +08:00
test_device_mesh.py [device] add DeviceMesh class to support logical device layout (#1394) 2022-08-02 19:23:48 +08:00
test_extract_alpha_beta.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_init_logical_pg.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_search_logical_device_mesh.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00