ColossalAI/colossalai/kernel
Jiarui Fang 1cb532ffec
[builder] multihead attn runtime building (#2203)
* [hotfix] correcnt cpu_optim runtime compilation

* [builder] multihead attn

* fix bug

* fix a bug
2022-12-27 16:06:09 +08:00
..
cuda_native [builder] multihead attn runtime building (#2203) 2022-12-27 16:06:09 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
op_builder [builder] multihead attn runtime building (#2203) 2022-12-27 16:06:09 +08:00
__init__.py [builder] multihead attn runtime building (#2203) 2022-12-27 16:06:09 +08:00