Skip to content

Commit 46ddb35

Browse files
Relax flaky requirement of host_dt > device_dt
1 parent 333db85 commit 46ddb35

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tests/test_sycl_kernel_submit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
8787
q.submit(axpyKernel, args, r).wait()
8888
ref_c = a * np.array(d, dtype=dtype) + b
8989
host_dt, device_dt = timer.dt
90-
assert host_dt > device_dt
90+
assert type(host_dt) is float and type(device_dt) is float
9191
assert np.allclose(c, ref_c), "Failed for {}".format(r)
9292

9393
for gr, lr in (
@@ -106,5 +106,5 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
106106
q.submit(axpyKernel, args, gr, lr, [dpctl.SyclEvent()]).wait()
107107
ref_c = a * np.array(d, dtype=dtype) + b
108108
host_dt, device_dt = timer.dt
109-
assert host_dt > device_dt
109+
assert type(host_dt) is float and type(device_dt) is float
110110
assert np.allclose(c, ref_c), "Faled for {}, {}".formatg(r, lr)

0 commit comments

Comments
 (0)