LuGY cbac782254 [zero]fix zero ckptIO with offload (#4529) ... * fix zero ckptio with offload * fix load device * saved tensors in ckpt should be on CPU * fix unit test * fix unit test * add clear cache * save memory for CI		2023-09-01 17:41:19 +08:00
..
test_gemini_checkpoint_io.py	[gemini] improve compatibility and add static placement policy (#4479)	2023-08-24 09:29:25 +08:00
test_gemini_torch_compability.py	[gemini] improve compatibility and add static placement policy (#4479)	2023-08-24 09:29:25 +08:00
test_general_checkpoint_io.py	[test] fixed codefactor format report (#4026)	2023-06-16 18:23:02 +08:00
test_low_level_zero_checkpoint_io.py	[zero]fix zero ckptIO with offload (#4529)	2023-09-01 17:41:19 +08:00
test_torch_ddp_checkpoint_io.py	[checkpointio] sharded optimizer checkpoint for DDP plugin (#4002)	2023-06-16 14:14:05 +08:00
test_torch_fsdp_checkpoint_io.py	[booster] torch fsdp fix ckpt (#3788)	2023-05-23 16:58:45 +08:00
utils.py	[plugin] a workaround for zero plugins' optimizer checkpoint (#3780)	2023-05-19 19:42:31 +08:00