diff now uses not_equal when the input is boolean

ndgrigorian · ndgrigorian · commit b280036a82cb · 2024-07-31T13:01:01.000-07:00
diff --git a/dpctl/tensor/_utility_functions.py b/dpctl/tensor/_utility_functions.py
@@ -464,7 +464,8 @@ def diff(x, /, *, axis=-1, n=1, prepend=None, append=None):
         slice(None) if i != axis else slice(None, -1) for i in range(x_nd)
     )
 
+    diff_op = dpt.not_equal if x.dtype == dpt.bool else dpt.subtract
     for _ in range(n):
-        arr = arr[sl0] - arr[sl1]
+        arr = diff_op(arr[sl0], arr[sl1])
 
     return arr

Original file line number	Diff line number	Diff line change
`@@ -464,7 +464,8 @@ def diff(x, /, *, axis=-1, n=1, prepend=None, append=None):`
`464`	`464`	`slice(None) if i != axis else slice(None, -1) for i in range(x_nd)`
`465`	`465`	`)`
`466`	`466`
	`467`	`+ diff_op = dpt.not_equal if x.dtype == dpt.bool else dpt.subtract`
`467`	`468`	`for _ in range(n):`
`468`		`- arr = arr[sl0] - arr[sl1]`
	`469`	`+ arr = diff_op(arr[sl0], arr[sl1])`
`469`	`470`
`470`	`471`	`return arr`