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
2025-11-13 06:11:09 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
2c0c85d3d337c43748ddd3ea23dfd2417de79f11
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
shenggan
5c3843dc98
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
..
csrc
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
__init__.py
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
builder.py
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
layer_norm.py
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
multihead_attention.py
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
scaled_softmax.py
add colossalai kernel module (
#55
)
2021-12-21 12:19:52 +08:00
