remove excess assert_raises_cudnn_disabled usage

apache · Aug 3, 2018 · 74bda2b · 74bda2b
1 parent d13a2d4
commit 74bda2b
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/tests/python/unittest/test_gluon_rnn.py b/tests/python/unittest/test_gluon_rnn.py
@@ -25,7 +25,6 @@
 from common import assert_raises_cudnn_disabled
 
 
-@assert_raises_cudnn_disabled()
 def test_rnn():
     cell = gluon.rnn.RNNCell(100, prefix='rnn_')
     inputs = [mx.sym.Variable('rnn_t%d_data'%i) for i in range(3)]
@@ -51,7 +50,6 @@ def test_lstm():
     assert outs == [(10, 100), (10, 100), (10, 100)]
 
 
-@assert_raises_cudnn_disabled()
 def test_lstm_forget_bias():
     forget_bias = 2.0
     stack = gluon.rnn.SequentialRNNCell()
@@ -89,7 +87,6 @@ def test_lstm_cpu_inference():
                                       rtol=1e-3, atol=1e-5)
 
 
-@assert_raises_cudnn_disabled()
 def test_gru():
     cell = gluon.rnn.GRUCell(100, prefix='rnn_')
     inputs = [mx.sym.Variable('rnn_t%d_data'%i) for i in range(3)]