diff --git a/tests/python/gpu/test_gluon_gpu.py b/tests/python/gpu/test_gluon_gpu.py index 9eeeec749211..a4132352fa26 100644 --- 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)