ColossalAI/colossalai/kernel/cuda_native
xcnick e0da01ea71
[hotfix] fix build error when torch version >= 1.13 (#1803)
2022-11-08 09:40:24 +08:00
..
csrc [hotfix] fix build error when torch version >= 1.13 (#1803) 2022-11-08 09:40:24 +08:00
__init__.py
flash_attention.py [kernel] more flexible flashatt interface (#1804) 2022-11-07 17:02:09 +08:00
layer_norm.py [NFC] polish colossalai/kernel/cuda_native/layer_norm.py code style (#980) 2022-05-17 10:25:06 +08:00
multihead_attention.py
scaled_softmax.py