.wip

Author: ricardoV94

pytensor/tensor/elemwise.py

@@ -1629,6 +1629,7 @@ def _c_all(self, node, name, input_names, output_names, sub):
 
     def c_code(self, node, name, inames, onames, sub):
         code = "\n".join(self._c_all(node, name, inames, onames, sub))
+        print(code)
         return code
 
     def c_headers(self, **kwargs):

tests/tensor/test_elemwise.py

@@ -400,21 +400,21 @@ def test_same_inputs(self):
 class TestCAReduce(unittest_tools.InferShapeTester):
     op = CAReduce
     cases = [
-        ((5, 6), None),
-        ((5, 6), (0, 1)),
-        ((5, 6), (0,)),
-        ((5, 6), (1,)),
+        # ((5, 6), None),
+        # ((5, 6), (0, 1)),
+        # ((5, 6), (0,)),
+        # ((5, 6), (1,)),
         ((5, 6), (-1,)),
-        ((5, 6), (-2,)),
-        ((5, 6), ()),
-        ((2, 3, 4, 5), (0, 1, 3)),
-        ((2, 3, 4, 5), (-2, -3)),
-        ((5, 0), None),
-        ((5, 0), (0,)),
+        # ((5, 6), (-2,)),
+        # ((5, 6), ()),
+        # ((2, 3, 4, 5), (0, 1, 3)),
+        # ((2, 3, 4, 5), (-2, -3)),
+        # ((5, 0), None),
+        # ((5, 0), (0,)),
         ((5, 0), (1,)),
-        ((5, 0), ()),
-        ((), None),
-        ((), ()),
+        # ((5, 0), ()),
+        # ((), None),
+        # ((), ()),
     ]
     type = TensorType
 
@@ -427,7 +427,8 @@ def with_mode(
         test_nan=False,
         tensor_op=None,
     ):
-        for xsh, tosum in self.cases:
+        for i, (xsh, tosum) in enumerate(self.cases):
+            print(i, xsh, tosum)
             if dtype == "floatX":
                 dtype = pytensor.config.floatX
             x = self.type(
@@ -530,7 +531,9 @@ def with_mode(
             else:
                 f_xv = f(xv)
                 assert f_xv.shape == zv.shape, (f_xv, zv)
+                print(xv, f_xv, zv)
                 utt.assert_allclose(zv, f_xv)
+            continue
 
             x = self.type(
                 dtype, shape=tuple(entry if entry == 1 else None for entry in xsh)