Merge pull request #1565 from IntelPython/boolean-reduction-always-allocate-reduction-temporary-as-usm-device

Boolean reductions allocate int32 temporary as USM-device

Author: oleksandr-pavlyk

dpctl/tensor/_utility_functions.py

@@ -36,10 +36,13 @@ def _boolean_reduction(x, axis, keepdims, func):
     res_usm_type = x.usm_type
 
     wait_list = []
+    # always allocate the temporary as
+    # int32 and usm-device  to ensure that atomic updates
+    # are supported
     res_tmp = dpt.empty(
         res_shape,
         dtype=dpt.int32,
-        usm_type=res_usm_type,
+        usm_type="device",
         sycl_queue=exec_q,
     )
     hev0, ev0 = func(