Move ref_c out of context manager

IntelPython · Jun 27, 2024 · 6f0a412 · 6f0a412
1 parent 8b40463
commit 6f0a412
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dpctl/tests/test_sycl_kernel_submit.py b/dpctl/tests/test_sycl_kernel_submit.py
@@ -76,7 +76,7 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
 
     b_np = dpt.asnumpy(b)
     a_np = dpt.asnumpy(a)
-    q.wait()
+
     for r in (
         [
             n_elems,
@@ -88,7 +88,7 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor):
         timer = dpctl.SyclTimer()
         with timer(q):
             q.submit(axpyKernel, args, r).wait()
-            ref_c = a_np * np.array(d, dtype=dtype) + b_np
+        ref_c = a_np * np.array(d, dtype=dtype) + b_np
         host_dt, device_dt = timer.dt
         assert type(host_dt) is float and type(device_dt) is float
         c_np = dpt.asnumpy(c)