ColossalAI/tests/test_zero
Edenzzzz 79f7a7b211
[misc] Accelerate CI for zero and dist optim (#5758)
* remove fp16 from lamb

* remove d2h copy in checking states

---------

Co-authored-by: Edenzzzz <wtan45@wisc.edu>
2024-06-05 11:25:19 +08:00
..
test_gemini [misc] Accelerate CI for zero and dist optim (#5758) 2024-06-05 11:25:19 +08:00
test_low_level [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00