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_utils
YuliangLiu0306 045afa3ea2
[hotfix] skip torchaudio tracing test (#3211)
2 years ago
..
test_checkpoint
test_checkpoint_io
test_lazy_init [hotfix] skip torchaudio tracing test (#3211) 2 years ago
test_activation_checkpointing.py
test_colo_checkpoint.py
test_commons.py
test_flash_attention.py updated flash attention usage 2 years ago
test_lazy_init_ctx.py
test_memory.py
test_norm_gradient_clipping.py
test_zero_gradient_clippling.py