mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-05 19:13:01 +00:00
[refactor] move process group from _DistSpec to ColoTensor. (#1203)
This commit is contained in:
@@ -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()
|
||||
|
Reference in New Issue
Block a user