Jiarui Fang ae7d3f4927 [refactor] move process group from _DistSpec to ColoTensor. (#1203)

2022-07-06 16:15:16 +08:00

..

__init__.py

fix import error (#880)

2022-04-26 19:28:40 +08:00

colo_init_context.py

[refactor] move process group from _DistSpec to ColoTensor. (#1203)

2022-07-06 16:15:16 +08:00

lazy_init_context.py

[context]support arbitary module materialization. (#1193)

2022-07-04 10:12:02 +08:00

utils.py

[context]use meta tensor to init model lazily. (#1187)

2022-06-29 21:02:30 +08:00