github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-24 14:33:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1cb532ffeccc44cd8afe08a740edbb226ce6c30c
ColossalAI/colossalai/kernel/op_builder
History
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
..
__init__.py
[builder] multihead attn runtime building (#2203)
2022-12-27 16:06:09 +08:00
builder.py
[builder] multihead attn runtime building (#2203)
2022-12-27 16:06:09 +08:00
cpu_adam.py
[hotfix] correcnt cpu_optim runtime compilation (#2197)
2022-12-26 16:45:14 +08:00
fused_optim.py
[builder] multihead attn runtime building (#2203)
2022-12-27 16:06:09 +08:00
multi_head_attn.py
[builder] multihead attn runtime building (#2203)
2022-12-27 16:06:09 +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