[fx] recovered skipped pipeline tests (#1338)

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)

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)