github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 06:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
34ca324b0d193623c89d8aea1aedb3c00ac2f654
ColossalAI/colossalai/kernel/cuda_native
History
Frank Lee 918bc94b6b [triton] added copyright information for flash attention (#2835) 
* [triton] added copyright information for flash attention

* polish code
2023-02-21 11:25:57 +08:00
..
csrc
…
__init__.py
…
flash_attention.py
[triton] added copyright information for flash attention (#2835)
2023-02-21 11:25:57 +08:00
layer_norm.py
…
multihead_attention.py
…
scaled_softmax.py
…