ColossalAI/colossalai/kernel/jit
アマデウス 4268ae017b
[kernel] added jit warmup (#1792)
2022-11-08 16:22:23 +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 [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00