ColossalAI/colossalai/kernel/cuda_native
Camille Zhong cd6a962e66 [NFC] polish code style (#4799) 2023-10-07 13:36:52 +08:00
..
csrc [NFC] polish code style (#4799) 2023-10-07 13:36:52 +08:00
mha [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
layer_norm.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multihead_attention.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
scaled_softmax.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00