mirror of https://github.com/hpcaitech/ColossalAI
c4b1b65931
* [test] fixed tests failed due to dtensor change * polish code |
||
---|---|---|
.. | ||
test_autochunk_alphafold | ||
test_autochunk_diffuser | ||
test_autochunk_transformer | ||
test_autochunk_vit |