ColossalAI/colossalai/kernel/cuda_native
yuxuan-lou 5f6ab35d25 Hotfix/format (#1274)
* [NFC] Polish colossalai/kernel/cuda_native/csrc/multi_tensor_lamb.cu code style. (#937)

* [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style

* [NFC] polish colossalai/kernel/cuda_native/csrc/scaled_masked_softmax.cpp code style

Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>
2022-07-13 12:08:21 +08:00
..
csrc Hotfix/format (#1274) 2022-07-13 12:08:21 +08:00
__init__.py refactor kernel (#142) 2022-01-13 16:47:17 +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 [formart] format fixed for kernel\cuda_native codes (#335) 2022-03-11 15:50:28 +08:00
scaled_softmax.py [NFC] polish colossalai/kernel/cuda_native/scaled_softmax.py code style (#955) 2022-05-17 10:25:06 +08:00