Fix cudnn tests

theano/sandbox/cuda/tests/test_dnn.py

@@ -121,7 +121,7 @@ def test_dnn_tag():
             [x],
             max_pool_2d(x, ds=(2, 2)),
             mode=mode_with_gpu.including("cudnn"))
-    except RuntimeError, e:
+    except (AssertionError, RuntimeError), e:
         assert not cuda.dnn.dnn_available()
         raised = True
     finally:

theano/sandbox/cuda/tests/test_nnet.py

@@ -301,6 +301,12 @@ class test_SoftMax(unittest.TestCase):
         self._cmp(2 << 15, 5, f, f_gpu)
         self._cmp(0, 10, f, f_gpu)
 
+    def test_softmax_cudnn(self):
+        if not cuda.dnn.dnn_available():
+            raise SkipTest(cuda.dnn.dnn_available.msg)
+        x = T.fmatrix('x')
+        z = T.nnet.softmax
+
         def check_types_with_cudnn(graph, graph_gpu):
             self._check_types(
                 graph,
@@ -320,7 +326,7 @@ class test_SoftMax(unittest.TestCase):
             check_types_with_cudnn
         )
 
-    def test_cudnn_softmax(self):
+    def test_cudnn_softmax_grad(self):
         if not cuda.dnn.dnn_available():
             raise SkipTest(cuda.dnn.dnn_available.msg)