ColossalAI/colossalai/kernel/cuda_native
ver217 12b4887097
[hotfix] fix CPUAdam kernel nullptr (#1410)
2022-08-05 19:45:45 +08:00
..
csrc [hotfix] fix CPUAdam kernel nullptr (#1410) 2022-08-05 19:45:45 +08:00
__init__.py
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 [NFC] polish colossalai/kernel/cuda_native/scaled_softmax.py code style (#955) 2022-05-17 10:25:06 +08:00