Skip to content

Commit ce39457

Browse files
Fixed test to pass on Iris Xe
1 parent 20cedd6 commit ce39457

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

dpctl/tests/test_tensor_asarray.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,11 +177,17 @@ def test_asarray_scalars():
177177
Y = dpt.asarray(5)
178178
assert Y.dtype == np.dtype(int)
179179
Y = dpt.asarray(5.2)
180-
assert Y.dtype == np.dtype(float)
180+
if Y.sycl_device.has_aspect_fp64:
181+
assert Y.dtype == np.dtype(float)
182+
else:
183+
assert Y.dtype == np.dtype(np.float32)
181184
Y = dpt.asarray(np.float32(2.3))
182185
assert Y.dtype == np.dtype(np.float32)
183186
Y = dpt.asarray(1.0j)
184-
assert Y.dtype == np.dtype(complex)
187+
if Y.sycl_device.has_aspect_fp64:
188+
assert Y.dtype == np.dtype(complex)
189+
else:
190+
assert Y.dtype == np.dtype(np.complex64)
185191
Y = dpt.asarray(ctypes.c_int(8))
186192
assert Y.dtype == np.dtype(ctypes.c_int)
187193

0 commit comments

Comments
 (0)