From f34431d776abb4cb84e7902b5dfc616aaab7a874 Mon Sep 17 00:00:00 2001
From: Sarbojit2019 <52527887+SarbojitAMD@users.noreply.github.com>
Date: Tue, 11 Jan 2022 13:52:13 +0530
Subject: [PATCH] SWDEV-3161551 - Fix for P2P sample failure (#2456)

[ROCm/hip-tests commit: 33ac9de928cc790d3aeab4bc27e4dd8ef3f9cf29]
---
 projects/hip-tests/samples/2_Cookbook/8_peer2peer/peer2peer.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/hip-tests/samples/2_Cookbook/8_peer2peer/peer2peer.cpp b/projects/hip-tests/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
index c2de71ce40..4ec6372108 100644
--- a/projects/hip-tests/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
+++ b/projects/hip-tests/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
@@ -174,6 +174,7 @@ int main() {
                     dim3(WIDTH / THREADS_PER_BLOCK_X, WIDTH / THREADS_PER_BLOCK_Y),
                     dim3(THREADS_PER_BLOCK_X, THREADS_PER_BLOCK_Y), 0, 0, gpuTransposeMatrix[0],
                     data[0], width);
+    HIPCHECK(hipDeviceSynchronize());
 
     HIPCHECK(hipSetDevice(peerGpu));
     TransposeMatrix[1] = (float*)malloc(NUM * sizeof(float));