Fix TrueDiv gradient for integer inputs

tanish1729 · Oct 8, 2024 · cdae903 · cdae903
1 parent aa616e6
commit cdae903
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pytensor/scalar/basic.py b/pytensor/scalar/basic.py
@@ -2036,7 +2036,10 @@ def grad(self, inputs, gout):
         # to the output; x/y is still a function of x
         # and y; it's just a step function.
         if all(a.dtype in discrete_dtypes for a in (x, y)):
-            return [x.zeros_like(), y.zeros_like()]
+            return [
+                x.zeros_like(dtype=config.floatX),
+                y.zeros_like(dtype=config.floatX),
+            ]
 
         first_part = gz / y