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-04-26 01:35:21 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
827a0af8ccdfd0165fba4ff7665567793aaf2b91
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
ver217
090f14fd6b
[misc] add reference (
#2930
)
...
* [misc] add reference * [misc] add license
2023-02-28 18:07:24 +08:00
..
csrc
[misc] add reference (
#2930
)
2023-02-28 18:07:24 +08:00
__init__.py
[kernel] fixed repeated loading of kernels (
#2549
)
2023-02-03 09:47:13 +08:00
flash_attention.py
[triton] added copyright information for flash attention (
#2835
)
2023-02-21 11:25:57 +08:00
layer_norm.py
[kernel] fixed repeated loading of kernels (
#2549
)
2023-02-03 09:47:13 +08:00
multihead_attention.py
[setup] support pre-build and jit-build of cuda kernels (
#2374
)
2023-01-06 20:50:26 +08:00
scaled_softmax.py
[misc] add reference (
#2930
)
2023-02-28 18:07:24 +08:00