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-19 00:55:09 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
20da6e48c8ebd7e17f68308b0bc45e977ff80471
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
ver217
e4f555f29a
[optim] refactor fused sgd (
#1134
)
2022-06-20 11:19:38 +08:00
..
csrc
[optim] refactor fused sgd (
#1134
)
2022-06-20 11:19:38 +08:00
__init__.py
refactor kernel (
#142
)
2022-01-13 16:47:17 +08:00
layer_norm.py
[NFC] polish colossalai/kernel/cuda_native/layer_norm.py code style (
#980
)
2022-05-17 10:25:06 +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
[NFC] polish colossalai/kernel/cuda_native/scaled_softmax.py code style (
#955
)
2022-05-17 10:25:06 +08:00
