Skip to content

Commit bb6db13

Browse files
Skip prod tests for complex output types on Gen9
1 parent 662bc45 commit bb6db13

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

dpctl/tests/test_tensor_sum.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -245,13 +245,15 @@ def test_prod_arg_out_dtype_matrix(arg_dtype, out_dtype):
245245

246246
out_dtype = dpt.dtype(out_dtype)
247247
arg_dtype = dpt.dtype(arg_dtype)
248-
if dpt.isdtype(out_dtype, "complex floating") and du._is_gen9(
249-
q.sycl_device
250-
):
251-
pytest.skip(
252-
"Product reduction for complex output are known "
253-
"to fail for Gen9 with 2024.0 compiler"
248+
if dpt.isdtype(out_dtype, "complex floating"):
249+
device_mask = (
250+
du.intel_device_info(q.sycl_device).get("device_id", 0) & 0xFF00
254251
)
252+
if device_mask in [0x3E00, 0x9B00]:
253+
pytest.skip(
254+
"Product reduction for complex output are known "
255+
"to fail for Gen9 with 2024.0 compiler"
256+
)
255257

256258
m = dpt.ones(100, dtype=arg_dtype)
257259
r = dpt.prod(m, dtype=out_dtype)

0 commit comments

Comments
 (0)