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-15 23:23:11 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
adebb3e04134393120ff32f6adda9b5ff477a993
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
ExtremeViscent
eaac03ae1d
[formart] format fixed for kernel\cuda_native codes (
#335
)
2022-03-11 15:50:28 +08:00
..
csrc
[zero] cpu adam kernel (
#288
)
2022-03-11 15:50:28 +08:00
__init__.py
refactor kernel (
#142
)
2022-01-13 16:47:17 +08:00
layer_norm.py
[formart] format fixed for kernel\cuda_native codes (
#335
)
2022-03-11 15:50:28 +08:00
multihead_attention.py
[formart] format fixed for kernel\cuda_native codes (
#335
)
2022-03-11 15:50:28 +08:00
scaled_softmax.py
Fixed docstring in colossalai (
#171
)
2022-01-21 10:44:30 +08:00
