including/excluding 'unsafe' modes added, 'unsafe' registered in tensor/opt, tests updated

theano/sandbox/cuda/tests/test_opt.py

@@ -59,10 +59,14 @@ def test_local_assert():
 def test_local_remove_all_assert():
     x = theano.tensor.fmatrix()
     a = theano.tensor.opt.assert_op(x, theano.tensor.eq(x, 0).any())
-    f = theano.function([x], a, mode=mode_with_gpu)
+    f = theano.function([x], a, mode=mode_with_gpu.including('unsafe'))
     topo = f.maker.fgraph.toposort()
     a_op = [n for n in topo if isinstance(n.op, theano.tensor.opt.Assert)]
     assert len(a_op) == 0
+    f = theano.function([x], a, mode=mode_with_gpu.excluding('unsafe'))
+    topo = f.maker.fgraph.toposort()
+    a_op = [n for n in topo if isinstance(n.op, theano.tensor.opt.Assert)]
+    assert len(a_op) == 1
 
 
 def test_int_pow():

theano/sandbox/gpuarray/tests/test_opt.py

@@ -28,10 +28,14 @@ def test_local_assert():
 def test_local_remove_all_assert():
     x = theano.tensor.fmatrix()
     a = theano.tensor.opt.assert_op(x, theano.tensor.eq(x, 0).any())
-    f = theano.function([x], a, mode=mode_with_gpu)
+    f = theano.function([x], a, mode=mode_with_gpu.including('unsafe'))
     topo = f.maker.fgraph.toposort()
     a_op = [n for n in topo if isinstance(n.op, theano.tensor.opt.Assert)]
     assert len(a_op) == 0
+    f = theano.function([x], a, mode=mode_with_gpu.excluding('unsafe'))
+    topo = f.maker.fgraph.toposort()
+    a_op = [n for n in topo if isinstance(n.op, theano.tensor.opt.Assert)]
+    assert len(a_op) == 1
 
 
 def test_flatten():

theano/tensor/opt.py

@@ -1670,12 +1670,15 @@ def local_remove_all_assert(node):
 # Disabled by default
 compile.optdb['canonicalize'].register('local_remove_all_assert',
                                        local_remove_all_assert,
+                                       'unsafe',
                                        use_db_name_as_tag=False)
 compile.optdb['stabilize'].register('local_remove_all_assert',
                                     local_remove_all_assert,
+                                    'unsafe',
                                     use_db_name_as_tag=False)
 compile.optdb['specialize'].register('local_remove_all_assert',
                                      local_remove_all_assert,
+                                     'unsafe',
                                      use_db_name_as_tag=False)
 
 

theano/tensor/tests/test_opt.py

@@ -3553,6 +3553,17 @@ class test_assert(utt.InferShapeTester):
         assert len(topo) == 1, topo
         assert topo[0].op == deep_copy_op, topo
 
+        mode = compile.mode.get_default_mode()
+        a = theano.tensor.opt.assert_op(x, T.eq(x, 0).any())
+        f = theano.function([x], a, mode=mode.including('unsafe'))
+        topo = f.maker.fgraph.toposort()
+        a_op = [n for n in topo if isinstance(n.op, T.opt.Assert)]
+        assert len(a_op) == 0
+        f = theano.function([x], a, mode=mode.excluding('unsafe'))
+        topo = f.maker.fgraph.toposort()
+        a_op = [n for n in topo if isinstance(n.op, T.opt.Assert)]
+        assert len(a_op) == 1
+
     def test_infer_shape(self):
 
         adscal = dscalar()