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-07-09 13:27:10 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
cea4292ae5ce7563f94bb7deef6f1b9bd155ecc5
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
ver217
f8a7148dec
[kernel] move all symlinks of kernel to
colossalai._C
(
#1971
)
2022-11-17 13:42:33 +08:00
..
csrc
[hotfix] fix build error when torch version >= 1.13 (
#1803
)
2022-11-08 09:40:24 +08:00
__init__.py
updated flash attention api
2022-11-15 15:25:39 +08:00
flash_attention.py
updated flash attention api
2022-11-15 15:25:39 +08:00
layer_norm.py
[kernel] move all symlinks of kernel to
colossalai._C
(
#1971
)
2022-11-17 13:42:33 +08:00
multihead_attention.py
[kernel] move all symlinks of kernel to
colossalai._C
(
#1971
)
2022-11-17 13:42:33 +08:00
scaled_softmax.py
[kernel] move all symlinks of kernel to
colossalai._C
(
#1971
)
2022-11-17 13:42:33 +08:00
