Skip to content

GitLab

"...Chat/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "e4fc57c3de6204ac66df75aa1752db1ec284f31f"
Unverified Commit f3ce7b83 authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[fx] recovered skipped pipeline tests (#1338)

parent 0c51ff2c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
tests/test_fx/test_comm_size_compute.py
View file @ f3ce7b83
...@@ -30,7 +30,6 @@ class MLP(torch.nn.Module): ...@@ -30,7 +30,6 @@ 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)
......
tests/test_fx/test_pipeline_passes.py
View file @ f3ce7b83
...@@ -39,7 +39,6 @@ def pipeline_pass_test_helper(model, data, pass_func): ...@@ -39,7 +39,6 @@ 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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment