Changed the RuntimeError to AssertionError and updated the test.

theano/sandbox/cuda/opt.py

@@ -455,7 +455,7 @@ def local_assert_no_cpu_op(node):
                 _logger.warning(("CPU op %s is detected in the computational"
                                  " graph") % node)
             elif config.assert_no_cpu_op == "raise":
-                raise RuntimeError("The op %s is on CPU." % node)
+                raise AssertionError("The op %s is on CPU." % node)
             elif config.assert_no_cpu_op == "pdb":
                 pdb.set_trace()
 
@@ -464,7 +464,7 @@ def local_assert_no_cpu_op(node):
 #Register the local_assert_no_cpu_op:
 assert_no_cpu_op = theano.tensor.opt.in2out(local_assert_no_cpu_op, name='assert_no_cpu_op')
 # 48.7 is after specialize device
-theano.compile.optdb.register('assert_no_cpu_op', assert_no_cpu_op, 48.7)
+theano.compile.optdb.register('assert_no_cpu_op', assert_no_cpu_op, 49.2)
 
 
 @register_opt()

theano/sandbox/cuda/tests/test_opt.py

@@ -96,7 +96,7 @@ def test_local_assert_no_cpu_op():
     numpy.random.seed(1)
     m = numpy.random.uniform(-1, 1, (10, 10)).astype("float32")
     ms = cuda.shared_constructor(m, name="m_shared")
-    out = theano.tensor.tanh(ms**2 + ms).dot(ms.T)
+    out = theano.tensor.tanh(ms).dot(ms.T)
 
     mode_local_assert = mode_with_gpu.including("assert_no_cpu_op")
     mode_local_assert = mode_local_assert.excluding("local_gpu_elemwise_0")
@@ -108,7 +108,7 @@ def test_local_assert_no_cpu_op():
     try:
         config.assert_no_cpu_op = 'raise'
 
-        assert_raises(RuntimeError, theano.function,
+        assert_raises(AssertionError, theano.function,
                         [], out, mode=mode_local_assert)
     finally:
         config.assert_no_cpu_op = old