From f3ce7b8336978414a95481ef76c7a0987c6f0cda Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Tue, 19 Jul 2022 09:49:50 +0800 Subject: [PATCH] [fx] recovered skipped pipeline tests (#1338) --- tests/test_fx/test_comm_size_compute.py | 1 - tests/test_fx/test_pipeline_passes.py | 1 - 2 files changed, 2 deletions(-) diff --git a/tests/test_fx/test_comm_size_compute.py b/tests/test_fx/test_comm_size_compute.py index f89d7c06a..c3bd78002 100644 --- a/tests/test_fx/test_comm_size_compute.py +++ b/tests/test_fx/test_comm_size_compute.py @@ -30,7 +30,6 @@ class MLP(torch.nn.Module): return x -@pytest.mark.skip('skip due to CI environment') def test_comm_size_compute(): model = MLP(MODEL_DIM) input_sample = torch.rand(BATCH_SIZE, MODEL_DIM) diff --git a/tests/test_fx/test_pipeline_passes.py b/tests/test_fx/test_pipeline_passes.py index 4d9e63d0d..de8a9402b 100644 --- a/tests/test_fx/test_pipeline_passes.py +++ b/tests/test_fx/test_pipeline_passes.py @@ -39,7 +39,6 @@ def pipeline_pass_test_helper(model, data, pass_func): assert output.equal(origin_output) -@pytest.mark.skip('skip due to CI environment') def test_pipeline_passes(): model = MLP(MODEL_DIM) data = torch.rand(BATCH_SIZE, MODEL_DIM)