ColossalAI/colossalai/kernel/jit
Frank Lee f3802d6b06
fixed jit default setting (#154)
2022-01-18 13:37:20 +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 fixed jit default setting (#154) 2022-01-18 13:37:20 +08:00