From 6d25416fb67ab2e2e93896a4ecafb49a03628d8f Mon Sep 17 00:00:00 2001 From: Lorenc Bushi Date: Thu, 8 Feb 2024 08:24:10 -0800 Subject: [PATCH] Fix access of device USM ptr from host --- sycl/test-e2e/KernelFusion/sync_two_queues_event_dep.cpp | 5 ++++- sycl/test-e2e/KernelFusion/sync_usm_mem_op.cpp | 8 +++++--- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/sycl/test-e2e/KernelFusion/sync_two_queues_event_dep.cpp b/sycl/test-e2e/KernelFusion/sync_two_queues_event_dep.cpp index 13b199c2bc0ba..79347114ec2a4 100644 --- a/sycl/test-e2e/KernelFusion/sync_two_queues_event_dep.cpp +++ b/sycl/test-e2e/KernelFusion/sync_two_queues_event_dep.cpp @@ -73,9 +73,12 @@ int main() { q1.wait(); q2.wait(); + int host_out[dataSize]; + q1.memcpy(host_out, out, dataSize * sizeof(int)); + q1.wait(); // Check the results for (size_t i = 0; i < dataSize; ++i) { - assert(out[i] == (40 * i * i) && "Computation error"); + assert(host_out[i] == (40 * i * i) && "Computation error"); } sycl::free(in1, q1); sycl::free(in2, q1); diff --git a/sycl/test-e2e/KernelFusion/sync_usm_mem_op.cpp b/sycl/test-e2e/KernelFusion/sync_usm_mem_op.cpp index a2899ff41d375..f3d2cc97b9ae5 100644 --- a/sycl/test-e2e/KernelFusion/sync_usm_mem_op.cpp +++ b/sycl/test-e2e/KernelFusion/sync_usm_mem_op.cpp @@ -61,14 +61,16 @@ int main() { "Queue should not be in fusion mode anymore"); fw.complete_fusion({ext::codeplay::experimental::property::no_barriers{}}); - + int host_out[dataSize]; + q.memcpy(host_out, out, dataSize * sizeof(int)); + q.wait(); for (size_t i = 0; i < dataSize; ++i) { - std::cout << out[i] << ", "; + std::cout << host_out[i] << ", "; } std::cout << "\n"; // Check the results for (size_t i = 0; i < dataSize; ++i) { - assert(out[i] == (20 * i * i) && "Computation error"); + assert(host_out[i] == (20 * i * i) && "Computation error"); assert(dst[i] == (5 * i) && "Computation error"); }