ColossalAI/tests/test_autochunk/test_autochunk_alphafold
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
..
benchmark_autochunk_alphafold.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_autochunk_alphafold_utils.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_autochunk_evoformer_block.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_autochunk_evoformer_stack.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00
test_autochunk_extramsa_block.py [test] refactor tests with spawn (#3452) 2023-04-06 14:51:35 +08:00