ColossalAI/colossalai/kernel/jit
shenggan 5c3843dc98
add colossalai kernel module (#55)
2021-12-21 12:19:52 +08:00
..
__init__.py add colossalai kernel module (#55) 2021-12-21 12:19:52 +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 add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00