tests/test_checkpoint_io/test_gemini_checkpoint_io.py · df63564184d5b13ab5f6edbca0c5e9abe211ef95 · OpenDAS / ColossalAI

"tests/test_ops/test_op.py" did not exist on "9bc5a77c31bec9c88c01ee7c739bc16a3b597e8b"

[gemini] support amp o3 for gemini (#4872) · df635641

Hongxin Liu authored Oct 12, 2023

* [gemini] support no reuse fp16 chunk

* [gemini] support no master weight for optim

* [gemini] support no master weight for gemini ddp

* [test] update gemini tests

* [test] update gemini tests

* [plugin] update gemini plugin

* [test] fix gemini checkpointio test

* [test] fix gemini checkpoint io

df635641

test_gemini_checkpoint_io.py 6.14 KB

Replace test_gemini_checkpoint_io.py