Disable GPU tests for the Dropout operator. (pytorch#75739)

Summary: Pull Request resolved: pytorch#75739 The tests are failing on platform010 and blocking the upgrade. Skip the tests given that Caffe2 on GPU is no longer supported. Test Plan: signals Reviewed By: ezyang Differential Revision: D35614159 Pull Request resolved: pytorch#75767 Approved by: https://github.com/ezyang
bowu1999 · Apr 14, 2022 · b311f25 · b311f25
1 parent 715e07b
commit b311f25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/caffe2/python/operator_test/dropout_op_test.py b/caffe2/python/operator_test/dropout_op_test.py
@@ -19,7 +19,7 @@ class TestDropout(serial.SerializedTestCase):
            in_place=st.booleans(),
            ratio=st.floats(0, 0.999),
            engine=st.sampled_from(["", "CUDNN"]),
-           **hu.gcs)
+           **hu.gcs_cpu_only)
     def test_dropout_is_test(self, X, in_place, ratio, engine, gc, dc):
         """Test with is_test=True for a deterministic reference impl."""
         # TODO(lukeyeager): enable this path when the GPU path is fixed
@@ -47,7 +47,7 @@ def reference_dropout_test(x):
            in_place=st.booleans(),
            output_mask=st.booleans(),
            engine=st.sampled_from(["", "CUDNN"]),
-           **hu.gcs)
+           **hu.gcs_cpu_only)
     @settings(deadline=10000)
     def test_dropout_ratio0(self, X, in_place, output_mask, engine, gc, dc):
         """Test with ratio=0 for a deterministic reference impl."""
@@ -80,7 +80,7 @@ def reference_dropout_ratio0(x):
            in_place=st.booleans(),
            output_mask=st.booleans(),
            engine=st.sampled_from(["", "CUDNN"]),
-           **hu.gcs)
+           **hu.gcs_cpu_only)
     @settings(deadline=10000)
     def test_dropout_ratio1(self, X, in_place, output_mask, engine, gc, dc):
         """Test with ratio=0 for a deterministic reference impl."""