..
kit
[Feature] Split cross-entropy computation in SP ( #5959 )
2024-09-10 12:06:50 +08:00
test_analyzer
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_auto_parallel
[test] Fix/fix testcase ( #5770 )
2024-06-03 15:26:01 +08:00
test_autochunk
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_booster
[plugin] support get_grad_norm ( #6115 )
2024-11-05 18:12:47 +08:00
test_checkpoint_io
[optim] hotfix adam load ( #6146 )
2024-11-20 16:36:37 +08:00
test_cluster
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_config
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
test_device
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_fp8
[fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 ( #6059 )
2024-09-14 10:40:01 +08:00
test_fx
[hotfix] fix testcase in test_fx/test_tracer ( #5779 )
2024-06-05 11:29:32 +08:00
test_infer
[release] update version ( #6041 )
2024-09-10 10:31:09 +08:00
test_lazy
[fp8] Merge feature/fp8_comm to main branch of Colossalai ( #6016 )
2024-08-22 09:21:34 +08:00
test_legacy
[chore] solve moe ckpt test failure and some other arg pass failure
2024-08-01 10:06:59 +08:00
test_lora
[Feature] Zigzag Ring attention ( #5905 )
2024-08-16 13:56:38 +08:00
test_moe
[hotfix] moe hybrid parallelism benchmark & follow-up fix ( #6048 )
2024-09-10 17:30:53 +08:00
test_optimizer
[MoE/ZeRO] Moe refactor with zero refactor ( #5821 )
2024-06-28 14:00:08 +08:00
test_pipeline
[Zerobubble] merge main. ( #6142 )
2024-11-19 19:00:36 +08:00
test_shardformer
[Zerobubble] merge main. ( #6142 )
2024-11-19 19:00:36 +08:00
test_smoothquant
[inference] Add smmoothquant for llama ( #4904 )
2023-10-16 11:28:44 +08:00
test_tensor
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_zero
[zero] support extra dp ( #6123 )
2024-11-12 11:20:46 +08:00
__init__.py
[zero] Update sharded model v2 using sharded param v2 ( #323 )
2022-03-11 15:50:28 +08:00