Disables failing tests due to cuDNN

apache · Mar 27, 2019 · b7e824c · b7e824c
1 parent 29578b5
commit b7e824c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python/gpu/test_gluon_gpu.py b/tests/python/gpu/test_gluon_gpu.py
@@ -158,6 +158,7 @@ def test_lstm_clip():
 
 @with_seed()
 @assert_raises_cudnn_not_satisfied(min_version='5.1.10')
+@unittest.skip("test fails due to cuDNN arch missmatch. temporarily disabled till it gets fixed. See https://github.com/apache/incubator-mxnet/issues/14502") 
 def test_rnn_layer():
     check_rnn_layer(gluon.rnn.RNN(100, num_layers=3))
     check_rnn_layer(gluon.rnn.RNN(100, activation='tanh', num_layers=3))
@@ -227,6 +228,7 @@ def test_rnn_layer_begin_state_type():
     modeling_layer(fake_data)
 
 
+@unittest.skip("test fails due to cuDNN arch missmatch. temporarily disabled till it gets fixed. See https://github.com/apache/incubator-mxnet/issues/14502") 
 def test_gluon_ctc_consistency():
     loss = mx.gluon.loss.CTCLoss()
     data = mx.nd.arange(0, 4, repeat=40, ctx=mx.gpu(0)).reshape((2,20,4)).flip(axis=0)