From 34131be8708270aa62e3b757b8325a11aeb3f71d Mon Sep 17 00:00:00 2001 From: FrankLeeeee Date: Tue, 19 Jul 2022 09:34:36 +0800 Subject: [PATCH] [fx] recovered skipped pipeline tests --- 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 f89d7c06af9a..c3bd78002ff2 100644 --- a/tests/test_fx/test_comm_size_compute.py +++ b/tests/test_fx/test_comm_size_compute.py @@ -30,7 +30,6 @@ def forward(self, x): 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 4d9e63d0d5a7..de8a9402ba56 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)