..
_C
[setup] support pre-build and jit-build of cuda kernels ( #2374 )
2023-01-06 20:50:26 +08:00
_analyzer
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
amp
[feature] Add clip_grad_norm for hybrid_parallel_plugin ( #4837 )
2023-10-12 11:32:37 +08:00
auto_parallel
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
autochunk
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
booster
[gemini] support gradient accumulation ( #4869 )
2023-10-17 14:07:21 +08:00
checkpoint_io
[checkpointio] hotfix torch 2.0 compatibility ( #4824 )
2023-10-07 10:45:52 +08:00
cli
[bug] Fix the version check bug in colossalai run when generating the cmd. ( #4713 )
2023-09-22 10:50:47 +08:00
cluster
[doc] polish shardformer doc ( #4779 )
2023-09-26 10:57:47 +08:00
context
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
device
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
fx
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
inference
[Refactor] Integrated some lightllm kernels into token-attention ( #4946 )
2023-10-19 22:22:47 +08:00
interface
[lazy] support from_pretrained ( #4801 )
2023-09-26 11:04:11 +08:00
kernel
[Refactor] Integrated some lightllm kernels into token-attention ( #4946 )
2023-10-19 22:22:47 +08:00
lazy
[doc] add lazy init docs ( #4808 )
2023-09-27 10:24:04 +08:00
legacy
[hotfix] fix torch 2.0 compatibility ( #4936 )
2023-10-18 11:05:25 +08:00
logging
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
nn
[test] add no master test for low level zero plugin ( #4934 )
2023-10-18 11:41:23 +08:00
pipeline
[format] applied code formatting on changed files in pull request 4820 ( #4886 )
2023-10-18 11:46:37 +08:00
shardformer
[nfc] fix some typo with colossalai/ docs/ etc. ( #4920 )
2023-10-18 15:44:04 +08:00
tensor
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
testing
[test] merge old components to test to model zoo ( #4945 )
2023-10-20 10:35:08 +08:00
utils
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
zero
[gemini] support gradient accumulation ( #4869 )
2023-10-17 14:07:21 +08:00
__init__.py
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
initialize.py
[misc] update pre-commit and run all files ( #4752 )
2023-09-19 14:20:26 +08:00
b8e770c832