..
_analyzer
[test] Fix/fix testcase ( #5770 )
2024-06-03 15:26:01 +08:00
_C
Clean up
2024-06-07 09:09:29 +00:00
accelerator
[hotfix] fix typo change MoECheckpintIO to MoECheckpointIO ( #5335 )
2024-03-05 21:52:30 +08:00
amp
[npu] change device to accelerator api ( #5239 )
2024-01-09 10:20:05 +08:00
auto_parallel
[pre-commit.ci] pre-commit autoupdate ( #5572 )
2024-07-01 17:16:41 +08:00
autochunk
[hotfix] Fix examples no pad token & auto parallel codegen bug; ( #5606 )
2024-04-18 18:15:50 +08:00
booster
[shardformer] optimize seq parallelism ( #6086 )
2024-10-11 13:44:40 +08:00
checkpoint_io
[colossalai/checkpoint_io/...] fix bug in load_state_dict_into_model; format error msg ( #6020 )
2024-09-02 16:56:35 +08:00
cli
[devops] fix extention building ( #5427 )
2024-03-05 15:35:54 +08:00
cluster
[FP8] rebase main ( #5963 )
2024-08-06 16:29:37 +08:00
context
[Fix]: implement thread-safety singleton to avoid deadlock for very large-scale training scenarios ( #5625 )
2024-04-25 14:45:52 +08:00
device
[Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor ( #5694 )
2024-05-14 13:52:45 +08:00
fx
[test] Fix/fix testcase ( #5770 )
2024-06-03 15:26:01 +08:00
inference
[shardformer] fix linear 1d row and support uneven splits for fused qkv linear ( #6084 )
2024-10-10 14:34:45 +08:00
interface
[Feature] Distributed optimizers: Lamb, Galore, CAME and Adafactor ( #5694 )
2024-05-14 13:52:45 +08:00
kernel
[pre-commit.ci] auto fixes from pre-commit.com hooks
2024-09-13 02:38:40 +00:00
lazy
[NPU]support npu ( #6089 )
2024-11-20 15:28:35 +08:00
legacy
[NPU]support npu ( #6089 )
2024-11-20 15:28:35 +08:00
logging
[fp8] Merge feature/fp8_comm to main branch of Colossalai ( #6016 )
2024-08-22 09:21:34 +08:00
moe
[hotfix] moe hybrid parallelism benchmark & follow-up fix ( #6048 )
2024-09-10 17:30:53 +08:00
nn
[misc] fix dist logger ( #5782 )
2024-06-05 15:04:22 +08:00
pipeline
[NPU]support npu ( #6089 )
2024-11-20 15:28:35 +08:00
quantization
[fp8] Disable all_gather intranode. Disable Redundant all_gather fp8 ( #6059 )
2024-09-14 10:40:01 +08:00
shardformer
[NPU]support npu ( #6089 )
2024-11-20 15:28:35 +08:00
tensor
[fp8] support fp8 amp for hybrid parallel plugin ( #5975 )
2024-08-07 18:21:08 +08:00
testing
[fp8] Merge feature/fp8_comm to main branch of Colossalai ( #6016 )
2024-08-22 09:21:34 +08:00
utils
Merge pull request #5310 from hpcaitech/feature/npu
2024-01-29 13:49:39 +08:00
zero
[NPU]support npu ( #6089 )
2024-11-20 15:28:35 +08:00
__init__.py
[devops] remove post commit ci ( #5566 )
2024-04-08 15:09:40 +08:00
initialize.py
[fp8] hotfix backward hook ( #6053 )
2024-09-11 16:11:25 +08:00
64f74a157e