Disable tests impacted by bugs in tvmop

apache · Mar 23, 2020 · b9bb072 · b9bb072
1 parent a57e5a2
commit b9bb072
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/python/unittest/test_deferred_compute.py b/tests/python/unittest/test_deferred_compute.py
@@ -324,6 +324,10 @@ def f(a, *, nd):
 
 
 def test_dc_simple_boolean_indexing():
+    if mx.test_utils.default_context() == mx.gpu(0) and mx.runtime.().is_enabled("TVM_OP"):
+        # Skip due to https://github.com/apache/incubator-mxnet/issues/17886
+        return
+
     def setup(*, nd):
         assert nd is mx.np
         x = mx.np.array([[0, 1], [1, 1], [2, 2]])
@@ -468,6 +472,10 @@ def forward(self, x):
 
 
 def test_dc_hybridblock_dynamic_shape():
+    if mx.test_utils.default_context() == mx.gpu(0) and mx.runtime.().is_enabled("TVM_OP"):
+        # Skip due to https://github.com/apache/incubator-mxnet/issues/17886
+        return
+
     class MyBlock(mx.gluon.HybridBlock):
         def __init__(self, *, prefix=None, params=None):
             super().__init__(prefix, params)