github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[hotfix] skip some unittest due to CI environment. (#1301)

pull/1308/head
YuliangLiu0306 2022-07-14 10:55:18 +08:00 committed by GitHub
parent 339520c6e0
commit 93a75433df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_fx/test_comm_size_compute.py
View File

@ -6,6 +6,7 @@ from torch.fx import symbolic_trace
from colossalai.fx.passes.meta_info_prop import MetaInfoProp from colossalai.fx.passes.meta_info_prop import MetaInfoProp
from colossalai.fx.passes.adding_split_node_pass import split_with_split_nodes_pass, uniform_split_pass from colossalai.fx.passes.adding_split_node_pass import split_with_split_nodes_pass, uniform_split_pass
from colossalai.fx.passes.utils import get_comm_size from colossalai.fx.passes.utils import get_comm_size
import pytest
MODEL_DIM = 16 MODEL_DIM = 16
BATCH_SIZE = 8 BATCH_SIZE = 8
@ -29,6 +30,7 @@ class MLP(torch.nn.Module):
return x return x
@pytest.mark.skip('skip due to CI environment')
def test_comm_size_compute(): def test_comm_size_compute():
model = MLP(MODEL_DIM) model = MLP(MODEL_DIM)
input_sample = torch.rand(BATCH_SIZE, MODEL_DIM) input_sample = torch.rand(BATCH_SIZE, MODEL_DIM)

2
tests/test_fx/test_pipeline_passes.py
View File

@ -5,6 +5,7 @@ import colossalai.nn as col_nn
from torch.fx import symbolic_trace from torch.fx import symbolic_trace
from colossalai.fx.passes.adding_split_node_pass import split_with_split_nodes_pass, balanced_split_pass, \ from colossalai.fx.passes.adding_split_node_pass import split_with_split_nodes_pass, balanced_split_pass, \
uniform_split_pass uniform_split_pass
import pytest
MODEL_DIM = 16 MODEL_DIM = 16
BATCH_SIZE = 8 BATCH_SIZE = 8
@ -37,6 +38,7 @@ def pipeline_pass_test_helper(model, data, pass_func):
assert output.equal(origin_output) assert output.equal(origin_output)
@pytest.mark.skip('skip due to CI environment')
def test_pipeline_passes(): def test_pipeline_passes():
model = MLP(MODEL_DIM) model = MLP(MODEL_DIM)
data = torch.rand(BATCH_SIZE, MODEL_DIM) data = torch.rand(BATCH_SIZE, MODEL_DIM)