tests/test_optimizer/test_adam_kernel.py · 4f68b3f10ce55a3563f943f8163b460d8c9fbb19 · OpenDAS / ColossalAI

[kernel] support pure fp16 for cpu adam and update gemini optim tests (#4921) · 4f68b3f1

Hongxin Liu authored Oct 16, 2023

* [kernel] support pure fp16 for cpu adam (#4896)

* [kernel] fix cpu adam kernel for pure fp16 and update tests (#4919)

* [kernel] fix cpu adam

* [test] update gemini optim test

4f68b3f1

test_adam_kernel.py 5.81 KB

Replace test_adam_kernel.py