This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-01-25 14:55:10 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
b699f54007c52b2f4ec56326a495b06858cf8856
ColossalAI
/
extensions
/
csrc
History
Steve Luo
b699f54007
optimize rmsnorm: add vectorized elementwise op, feat loop unrolling (
#5441
)
2024-03-12 17:48:02 +08:00
..
arm
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
common
optimize rmsnorm: add vectorized elementwise op, feat loop unrolling (
#5441
)
2024-03-12 17:48:02 +08:00
cuda
optimize rmsnorm: add vectorized elementwise op, feat loop unrolling (
#5441
)
2024-03-12 17:48:02 +08:00
x86
refactor code
2024-03-11 17:06:57 +08:00
__init__.py
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
scaled_softmax.py
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
