From 340dc9d98a1115218809feffe0514c2afefdf8a9 Mon Sep 17 00:00:00 2001 From: Sheng Zha Date: Tue, 31 Jul 2018 17:48:19 -0700 Subject: [PATCH] remove excess assert_raises_cudnn_disabled usage --- tests/python/unittest/test_gluon_rnn.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/python/unittest/test_gluon_rnn.py b/tests/python/unittest/test_gluon_rnn.py index d629169cf154..15f85003276f 100644 --- 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)]