ColossalAI/colossalai/kernel/cuda_native
ExtremeViscent 22d1df224d [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968)
code style
2022-05-17 10:25:06 +08:00
..
csrc [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968) 2022-05-17 10:25:06 +08:00
__init__.py
layer_norm.py Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +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