Match Array API tests (run array-api tests)

Quantco · Jul 10, 2024 · 5c5ce1d · 5c5ce1d
1 parent 550f803
commit 5c5ce1d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ndonnx/_core/_impl.py b/ndonnx/_core/_impl.py
@@ -112,9 +112,13 @@ def cosh(self, x):
         return _unary_op(x, opx.cosh, dtypes.float32)
 
     def divide(self, x, y):
-        return _variadic_op(
-            [x, y], opx.div, via_dtype=dtypes.float64, cast_return=False
+        x, y = promote(x, y)
+        via_dtype = (
+            dtypes.float64
+            if x.dtype in (dtypes.nint64, dtypes.int64)
+            else dtypes.float32
         )
+        return _variadic_op([x, y], opx.div, via_dtype=via_dtype, cast_return=False)
 
     def equal(self, x, y) -> Array:
         x, y = promote(x, y)