From 828e62fe4f84acf714966b96a86f4adaeb3d6859 Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Thu, 28 Feb 2019 00:56:07 +0530
Subject: [PATCH] Fix hipBusBW sample for P2P bidirectional test

---
 hipamd/samples/1_Utils/hipBusBandwidth/hipBusBandwidth.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/hipamd/samples/1_Utils/hipBusBandwidth/hipBusBandwidth.cpp b/hipamd/samples/1_Utils/hipBusBandwidth/hipBusBandwidth.cpp
index bbeddf2f48..9e61e018e8 100644
--- a/hipamd/samples/1_Utils/hipBusBandwidth/hipBusBandwidth.cpp
+++ b/hipamd/samples/1_Utils/hipBusBandwidth/hipBusBandwidth.cpp
@@ -812,6 +812,7 @@ void RunBenchmark_P2P_Bidir(ResultDatabase& resultDB) {
             hipSetDevice(currentGpu);
             hipMalloc((void**)&currentGpuMem[0], sizeof(float) * numMaxFloats);
             hipMalloc((void**)&currentGpuMem[1], sizeof(float) * numMaxFloats);
+            enablePeer2Peer(peerGpu,currentGpu);
 
             hipSetDevice(peerGpu);
             hipMalloc((void**)&peerGpuMem[0], sizeof(float) * numMaxFloats);
@@ -899,6 +900,7 @@ void RunBenchmark_P2P_Bidir(ResultDatabase& resultDB) {
             }
 
             disablePeer2Peer(currentGpu, peerGpu);
+            disablePeer2Peer(peerGpu, currentGpu);
 
             hipEventDestroy(start);
             hipEventDestroy(stop);