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
FrankLeeeee 0310b76e9d
Merge branch 'main' into sync/main
9 months ago
..
kit [workflow] fixed oom tests (#5275) 11 months ago
test_analyzer
test_auto_parallel
test_autochunk
test_booster [fix] remove unnecessary dp_size assert (#5351) 10 months ago
test_checkpoint_io [fsdp] impl save/load shard model/optimizer (#5357) 9 months ago
test_cluster
test_config
test_device
test_fx
test_gptq
test_infer [Inference]Move benchmark-related code to the example directory. (#5408) 9 months ago
test_lazy [workflow] fixed oom tests (#5275) 11 months ago
test_legacy
test_moe [moe] fix tests 10 months ago
test_optimizer [lr-scheduler] fix load state dict and add test (#5369) 10 months ago
test_pipeline Merge branch 'main' into sync/npu 10 months ago
test_shardformer [hotfix] Fix ShardFormer test execution path when using sequence parallelism (#5230) 10 months ago
test_smoothquant
test_tensor
test_utils [feat] refactored extension module (#5298) 10 months ago
test_zero
__init__.py