Add malloc_device call in QueueIDCheck

intel · Feb 10, 2024 · 81684de · 81684de
1 parent 9b2f696
commit 81684de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sycl/unittests/xpti_trace/QueueIDCheck.cpp b/sycl/unittests/xpti_trace/QueueIDCheck.cpp
@@ -141,8 +141,8 @@ TEST_F(QueueID, QueueCreationUSMOperations) {
  auto Queue0ImplPtr = sycl::detail::getSyclObjImpl(Q0);
  auto QueueIDSTr = std::to_string(Queue0ImplPtr->getQueueID());
 
- unsigned char *AllocSrc = (unsigned char *)sycl::malloc_shared(1, Q0);
- unsigned char *AllocDst = (unsigned char *)sycl::malloc_shared(1, Q0);
+ unsigned char *AllocSrc = (unsigned char *)sycl::malloc_device(1, Q0);
+ unsigned char *AllocDst = (unsigned char *)sycl::malloc_device(1, Q0);
  Q0.memset(AllocSrc, 42, 1).wait();
  checkTaskBeginEnd(QueueIDSTr);