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

This commit is contained in:
Jiarui Fang
2022-07-06 16:15:16 +08:00
committed by GitHub
parent 5da87ce35d
commit ae7d3f4927
34 changed files with 452 additions and 367 deletions

View File

@@ -1,7 +1,9 @@
import torch
from colossalai.fx.proxy import ColoProxy
import pytest
@pytest.mark.skip
def test_coloproxy():
# create a dummy node only for testing purpose
model = torch.nn.Linear(10, 10)
@@ -20,4 +22,4 @@ def test_coloproxy():
if __name__ == '__main__':
test_coloproxy()
test_coloproxy()