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
Wenhao Chen 196b85368b
[pipeline]: add p2p fallback order and fix interleaved pp deadlock (#5214)
11 months ago
..
kit [gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150) 12 months ago
test_analyzer
test_auto_parallel
test_autochunk
test_booster [plugin]fix 3d checkpoint load when booster boost without optimizer. (#5135) 1 year ago
test_checkpoint_io
test_cluster
test_config
test_device
test_fx
test_gptq
test_infer [Hotfix] Fix model policy matching strategy in ShardFormer (#5064) 1 year ago
test_infer_ops/triton
test_lazy
test_legacy
test_moe fix optim 11 months ago
test_optimizer
test_pipeline [pipeline]: add p2p fallback order and fix interleaved pp deadlock (#5214) 11 months ago
test_shardformer [pipeline]: fix p2p comm, add metadata cache and support llama interleaved pp (#5134) 11 months ago
test_smoothquant
test_tensor
test_utils
test_zero
__init__.py