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