ColossalAI/colossalai/kernel
Frank Lee 533d0c46d8
[kernel] fixed the include bug in dropout kernel (#999)
2022-05-18 21:43:18 +08:00
..
cuda_native [kernel] fixed the include bug in dropout kernel (#999) 2022-05-18 21:43:18 +08:00
jit [NFC] polish colossalai/kernel/jit/bias_gelu.py code style (#946) 2022-05-17 10:25:06 +08:00
__init__.py fix format (#568) 2022-04-06 11:40:59 +08:00