From ee00ebc42f1b19893970eb9c26e624d1cf24becb Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Wed, 18 Oct 2023 14:55:03 +0000
Subject: [PATCH] SWDEV-425568 - Do cpu wait for device sync.

Change-Id: Ia05e09bd10ab623b36c74037e9988e132120dd9f
---
 hipamd/src/hip_device_runtime.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/hipamd/src/hip_device_runtime.cpp b/hipamd/src/hip_device_runtime.cpp
index 3fbfce9d4e..a728a8f946 100644
--- a/hipamd/src/hip_device_runtime.cpp
+++ b/hipamd/src/hip_device_runtime.cpp
@@ -579,8 +579,8 @@ hipError_t hipDeviceSetSharedMemConfig(hipSharedMemConfig config) {
 
 hipError_t hipDeviceSynchronize() {
   HIP_INIT_API(hipDeviceSynchronize);
-  constexpr bool kDontWaitForCpu = false;
-  hip::Stream::SyncAllStreams(hip::getCurrentDevice()->deviceId(), kDontWaitForCpu);
+  constexpr bool kDoWaitForCpu = true;
+  hip::Stream::SyncAllStreams(hip::getCurrentDevice()->deviceId(), kDoWaitForCpu);
   HIP_RETURN(hipSuccess);
 }