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

* remove print
2022-12-28 19:45:49 +08:00
..
__init__.py [builder] multihead attn runtime building (#2203) 2022-12-27 16:06:09 +08:00
builder.py [builder] polish builder with better base class (#2216) 2022-12-28 19:45:49 +08:00
cpu_adam.py [builder] polish builder with better base class (#2216) 2022-12-28 19:45:49 +08:00
fused_optim.py [builder] polish builder with better base class (#2216) 2022-12-28 19:45:49 +08:00
multi_head_attn.py [builder] polish builder with better base class (#2216) 2022-12-28 19:45:49 +08:00
utils.py [buider] use builder() for cpu adam and fused optim in setup.py (#2187) 2022-12-23 16:05:13 +08:00