ColossalAI/tests/test_optimizer/test_fused_adam_kernel.py at 24246f7aa5c4a7efb043bd61fe5b00f272f278ef - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 09:52:50 +00:00

Files

Jiarui Fang 355ffb386e [builder] unified cpu_optim fused_optim inferface (#2190 )

2022-12-23 20:57:41 +08:00

2.5 KiB

Raw Blame History

View Raw