From b665052bd75f749c2ceebd7e43b5613c996607a0 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)

---
 samples/2_Cookbook/8_peer2peer/peer2peer.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/samples/2_Cookbook/8_peer2peer/peer2peer.cpp b/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
index c2de71ce40..4ec6372108 100644
--- a/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
+++ b/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));