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-04-25 09:12:10 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
1dedb57747270f32be5d0e67abc1ad2fff658f8f
ColossalAI
/
colossalai
/
kernel
History
yuehuayingxueluo
21ad4a27f9
[Inference/opt]Optimize the mid tensor of RMS Norm (
#5350
)
...
* opt rms_norm * fix bugs in rms_layernorm
2024-02-02 15:06:01 +08:00
..
jit
[npu] change device to accelerator api (
#5239
)
2024-01-09 10:20:05 +08:00
triton
[Inference/opt]Optimize the mid tensor of RMS Norm (
#5350
)
2024-02-02 15:06:01 +08:00
__init__.py
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
extensions
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
kernel_loader.py
[feat] refactored extension module (
#5298
)
2024-01-25 17:01:48 +08:00
