1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-02 13:45:36 +00:00
ColossalAI/tests
Wenxuan Tan ec73f1b5e2
[CI] Cleanup Dist Optim tests with shared helper funcs ()
* Refractor and cleanup using common helper funcs. Tests passed

* Update comments

* Fix relative import

* Fix param fetching bug
2025-02-12 13:42:34 +08:00
..
kit [CI] Cleanup Dist Optim tests with shared helper funcs () 2025-02-12 13:42:34 +08:00
test_analyzer [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_auto_parallel [test] Fix/fix testcase () 2024-06-03 15:26:01 +08:00
test_autochunk [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_booster [plugin] support get_grad_norm () 2024-11-05 18:12:47 +08:00
test_checkpoint_io [checkpointio] support non blocking pin load () 2024-12-25 17:03:25 +08:00
test_cluster [misc] refactor launch API and tensor constructor () 2024-04-29 10:40:11 +08:00
test_config [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
test_device [misc] refactor launch API and tensor constructor () 2024-04-29 10:40:11 +08:00
test_fp8 [fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 () 2024-09-14 10:40:01 +08:00
test_fx [hotfix] fix testcase in test_fx/test_tracer () 2024-06-05 11:29:32 +08:00
test_infer [release] update version () 2024-09-10 10:31:09 +08:00
test_lazy [fp8] Merge feature/fp8_comm to main branch of Colossalai () 2024-08-22 09:21:34 +08:00
test_legacy [FP8] rebase main () 2024-08-06 16:29:37 +08:00
test_lora [fp8] Merge feature/fp8_comm to main branch of Colossalai () 2024-08-22 09:21:34 +08:00
test_moe [hotfix] moe hybrid parallelism benchmark & follow-up fix () 2024-09-10 17:30:53 +08:00
test_optimizer [CI] Cleanup Dist Optim tests with shared helper funcs () 2025-02-12 13:42:34 +08:00
test_pipeline [hotfix] fix hybrid checkpointio for sp+dp () 2025-02-06 17:21:04 +08:00
test_shardformer [shardformer] support ep for deepseek v3 () 2025-02-11 16:10:25 +08:00
test_smoothquant [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00
test_tensor [misc] refactor launch API and tensor constructor () 2024-04-29 10:40:11 +08:00
test_zero [zero] support extra dp () 2024-11-12 11:20:46 +08:00
__init__.py [zero] Update sharded model v2 using sharded param v2 () 2022-03-11 15:50:28 +08:00
conftest.py [checkpointio] support load-pin overlap () 2025-01-07 16:16:04 +08:00