mirror of https://github.com/hpcaitech/ColossalAI
c4b1b65931
* [test] fixed tests failed due to dtensor change * polish code |
||
---|---|---|
.. | ||
test_alpha_beta.py | ||
test_device_mesh.py | ||
test_extract_alpha_beta.py | ||
test_init_logical_pg.py | ||
test_search_logical_device_mesh.py |