Hongxin Liu fc5cef2c79 [lazy] support init on cuda (#4269) ... * [lazy] support init on cuda * [test] update lazy init test * [test] fix transformer version		2023-07-19 16:43:01 +08:00
..
lazy_init_utils.py	[lazy] support init on cuda (#4269)	2023-07-19 16:43:01 +08:00
test_distribute.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00
test_models.py	[lazy] support init on cuda (#4269)	2023-07-19 16:43:01 +08:00