tests/test_tensor/test_zero_optim.py · 7d14b473f0013e62f2e2d9ba60ac8e39d62eb5fb · OpenDAS / ColossalAI

[gemini] gemini mgr supports "cpu" placement policy (#1118) · 7d14b473

ver217 authored Jun 15, 2022

* update gemini mgr

* update chunk

* add docstr

* polish placement policy

* update test chunk

* update test zero

* polish unit test

* remove useless unit test

7d14b473

test_zero_optim.py 4.24 KB

Replace test_zero_optim.py