ColossalAI/colossalai/kernel
Jiarui Fang 54de05da5d
[builder] polish builder with better base class (#2216)
* [builder] polish builder

* remove print
2022-12-28 19:45:49 +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] polish builder with better base class (#2216) 2022-12-28 19:45:49 +08:00
__init__.py [builder] multihead attn runtime building (#2203) 2022-12-27 16:06:09 +08:00