Skip test_lc_1d for ROCM (#50964)

arindamroy-eng · facebook-github-bot · commit 09b896261c66 · 2021-01-25T11:43:37.000-08:00
Summary: The test is flaky on ROCM when deadline is set to 1 second. This is affecting builds as it is failing randomly. Disabling for now. Signed-off-by: Arindam Roy <rarindam@gmail.com> Pull Request resolved: #50964 Reviewed By: houseroad Differential Revision: D26049370 Pulled By: BIT-silence fbshipit-source-id: 22337590a8896ad75f1281e56fbbeae897f5c3b2
diff --git a/caffe2/python/operator_test/locally_connected_op_test.py b/caffe2/python/operator_test/locally_connected_op_test.py
@@ -6,11 +6,12 @@
 from hypothesis import given, settings, assume
 import hypothesis.strategies as st
 
-from caffe2.python import core, utils
+from caffe2.python import core, utils, workspace
 import caffe2.python.hypothesis_test_util as hu
 import caffe2.python.serialized_test.serialized_test_util as serial
 
 
+
 class TestLocallyConnectedOp(serial.SerializedTestCase):
     @given(N=st.integers(1, 3),
            C=st.integers(1, 3),
@@ -104,6 +105,9 @@ def lc_2d_nhwc(X, W, b=None):
            **hu.gcs)
     @settings(deadline=1000)
     def test_lc_1d(self, N, C, size, M, kernel, op_name, use_bias, gc, dc):
+        if workspace.has_hip_support:
+            # Skip as test flaky on ROCM with deadline set to 1000
+            return
         if size < kernel:
             kernel = size
 
