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)