ColossalAI/colossalai/kernel
Jiarui Fang bc0e271e71
[buider] use builder() for cpu adam and fused optim in setup.py (#2187)
2022-12-23 16:05:13 +08:00
..
cuda_native updated attention kernel (#2133) 2022-12-16 10:54:03 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
op_builder [buider] use builder() for cpu adam and fused optim in setup.py (#2187) 2022-12-23 16:05:13 +08:00
__init__.py fix format (#568) 2022-04-06 11:40:59 +08:00