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_auto_parallel
Zihao 18dbe76cae
[auto-parallel] add auto-offload feature (#3154)
2 years ago
..
test_ckpt_solvers [hotfix] skip auto checkpointing tests (#3029) 2 years ago
test_offload [auto-parallel] add auto-offload feature (#3154) 2 years ago
test_pass
test_tensor_shard [autoparallel] apply repeat block to reduce solving time (#2912) 2 years ago
__init__.py