Hongxin Liu
|
070df689e6
|
[devops] fix extention building (#5427)
|
9 months ago |
Frank Lee
|
7cfed5f076
|
[feat] refactored extension module (#5298)
* [feat] refactored extension module
* polish
* polish
* polish
* polish
* polish
* polish
* polish
* polish
* polish
* polish
|
10 months ago |
Xuanlei Zhao
|
dd2c28a323
|
[npu] use extension for op builder (#5172)
* update extension
* update cpu adam
* update is
* add doc for cpu adam
* update kernel
* update commit
* update flash
* update memory efficient
* update flash attn
* update flash attention loader
* update api
* fix
* update doc
* update example time limit
* reverse change
* fix doc
* remove useless kernel
* fix
* not use warning
* update
* update
|
11 months ago |
Hongxin Liu
|
079bf3cb26
|
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit
* [misc] run pre-commit
* [misc] remove useless configuration files
* [misc] ignore cuda for clang-format
|
1 year ago |
Mashiro
|
cfa607080f
|
[Fix] Fix compile error (#4357)
|
1 year ago |
Yanming W
|
269150b6f4
|
[Docker] Fix a couple of build issues (#3691)
|
2 years ago |
digger-yu
|
b7141c36dd
|
[CI] fix some spelling errors (#3707)
* fix spelling error with examples/comminity/
* fix spelling error with tests/
* fix some spelling error with tests/ colossalai/ etc.
|
2 years ago |
Hongxin Liu
|
173dad0562
|
[misc] add verbose arg for zero and op builder (#3552)
* [misc] add print verbose
* [gemini] add print verbose
* [zero] add print verbose for low level
* [misc] add print verbose for op builder
|
2 years ago |
Yasyf Mohamedali
|
19fa0e57f6
|
Remove extraneous comma (#2993)
Prevents `TypeError: category must be a Warning subclass, not 'str'`.
|
2 years ago |
Frank Lee
|
3a5d93bc2c
|
[kernel] cached the op kernel and fixed version check (#2886)
* [kernel] cached the op kernel and fixed version check
* polish code
* polish code
|
2 years ago |
Frank Lee
|
40d376c566
|
[setup] support pre-build and jit-build of cuda kernels (#2374)
* [setup] support pre-build and jit-build of cuda kernels
* polish code
* polish code
* polish code
* polish code
* polish code
* polish code
|
2 years ago |
Frank Lee
|
9b765e7a69
|
[setup] removed the build dependency on colossalai (#2307)
|
2 years ago |
Jiarui Fang
|
bc0e271e71
|
[buider] use builder() for cpu adam and fused optim in setup.py (#2187)
|
2 years ago |