ColossalAI/colossalai/kernel/jit
luoling-LC 72c71b67ec [NFC] polish colossalai/kernel/jit/bias_gelu.py code style (#946)
Co-authored-by: jnbai <897086360@qq.com>
2022-05-17 10:25:06 +08:00
..
__init__.py fixed jit default setting (#154) 2022-01-18 13:37:20 +08:00
bias_dropout_add.py add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
bias_gelu.py [NFC] polish colossalai/kernel/jit/bias_gelu.py code style (#946) 2022-05-17 10:25:06 +08:00
option.py adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00