From f91029e50d0a735c19544e9ad4e3ee52d6eea37a Mon Sep 17 00:00:00 2001 From: Nikita Grigorian Date: Fri, 28 Jun 2024 00:21:28 +0000 Subject: [PATCH] Set ref_c prior to axpyKernel call --- dpctl/tests/test_sycl_kernel_submit.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dpctl/tests/test_sycl_kernel_submit.py b/dpctl/tests/test_sycl_kernel_submit.py index 8c9baed4f8..2bbe33e9a0 100644 --- a/dpctl/tests/test_sycl_kernel_submit.py +++ b/dpctl/tests/test_sycl_kernel_submit.py @@ -76,8 +76,7 @@ def test_create_program_from_source(ctype_str, dtype, ctypes_ctor): b_np = dpt.asnumpy(b) a_np = dpt.asnumpy(a) - print(b_np) - print(a_np) + ref_c = a_np * np.array(d, dtype=dtype) + b_np for r in ( [ @@ -90,7 +89,6 @@ 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 host_dt, device_dt = timer.dt assert type(host_dt) is float and type(device_dt) is float c_np = dpt.asnumpy(c)