BUGFIX: binary ops using function calls (e.g., logical_and, minimum) …

…were missing the function names when constructing deferred ops
Python-for-HPC · Jan 18, 2024 · 484ab3e · 484ab3e
1 parent 0373d4c
commit 484ab3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ramba/ramba.py b/ramba/ramba.py
@@ -6056,9 +6056,9 @@ def array_binop(
             #add_constraint([(self, list(range(1, self.ndim + 1))), (rhs, list(range(1, rhs.ndim + 1)))])
             # We might need replaced_args below due to the div-to-mul rewrite above.
             if inplace:
-                return DAGshape(new_shape, new_dtype, self, replaced_args=[self, rhs, op, optext])
+                return DAGshape(new_shape, new_dtype, self, replaced_args=[self, rhs, op, optext, opfunc])
             else:
-                return DAGshape(new_shape, new_dtype, False, replaced_args=[self, rhs, op, optext])
+                return DAGshape(new_shape, new_dtype, False, replaced_args=[self, rhs, op, optext, opfunc])
 
 
     @classmethod