You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_autochunk
Frank Lee c4b1b65931
[test] fixed tests failed due to dtensor change (#4082)
1 year ago
..
test_autochunk_alphafold [test] refactor tests with spawn (#3452) 2 years ago
test_autochunk_diffuser [test] fixed tests failed due to dtensor change (#4082) 1 year ago
test_autochunk_transformer [devops] update torch version of CI (#3725) 2 years ago
test_autochunk_vit [devops] update torch version of CI (#3725) 2 years ago