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 09:04:50 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
4df6471f5d044150edfedfe1cf8370a604e096ca
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
Jiarui Fang
e761ad2cd7
Revert "[zero] add ZeroTensorShardStrategy (
#793
)" (
#806
)
2022-04-19 14:40:02 +08:00
..
csrc
Revert "[zero] add ZeroTensorShardStrategy (
#793
)" (
#806
)
2022-04-19 14:40:02 +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