tests/test_zero/test_gemini/test_grad_clip.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_grad_clip.py 4.63 KB

Replace test_grad_clip.py