ColossalAI/colossalai/kernel
HELSON b31daed4cf
fix bugs in CPU adam (#633)
* add cpu adam counter for all cpu adam

* fixed updating error in adam kernel
2022-04-02 17:04:05 +08:00
..
cuda_native fix bugs in CPU adam (#633) 2022-04-02 17:04:05 +08:00
jit adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
__init__.py refactor kernel (#142) 2022-01-13 16:47:17 +08:00