ColossalAI/colossalai/kernel/jit
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
2022-01-20 13:44:51 +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 add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
option.py adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00