ColossalAI/tests
2025-07-14 07:50:38 +00:00
..
kit [CI] disable timm_regnetv_040 as aten::_unique2 is not supproted 2025-07-14 07:50:38 +00:00
test_analyzer
test_auto_parallel
test_autochunk
test_booster [plugin] support get_grad_norm (#6115) 2024-11-05 18:12:47 +08:00
test_checkpoint_io [shardformer] support pipeline for deepseek v3 and optimize lora save (#6188) 2025-02-14 14:48:54 +08:00
test_cluster [misc] update torch version (#6206) 2025-02-24 14:35:48 +08:00
test_config
test_device
test_fp8 [ci] update ci (#6254) 2025-04-18 16:40:53 +08:00
test_fx
test_infer [upgrade]Upgrade transformers (#6320) 2025-05-27 14:29:01 +08:00
test_lazy [fix] fix_lazy_init for deepseek model in transformers 2025-06-02 11:31:45 +08:00
test_legacy
test_lora
test_moe
test_optimizer [CI] Cleanup Dist Optim tests with shared helper funcs (#6125) 2025-02-12 13:42:34 +08:00
test_pipeline [hotfix] fix hybrid checkpointio for sp+dp (#6184) 2025-02-06 17:21:04 +08:00
test_shardformer [feat] support qwen3 in shardformer 2025-07-10 13:57:52 +08:00
test_smoothquant
test_tensor
test_zero release 2025-05-27 14:38:59 +08:00
__init__.py
conftest.py [checkpointio] support load-pin overlap (#6177) 2025-01-07 16:16:04 +08:00