Fix hipBusBW sample for P2P bidirectional test

[ROCm/hip-tests commit: 5f54c5cb59]
2019-02-28 00:56:07 +05:30
parent cfdf6ca042
commit 6cf575a749
1 changed files with 2 additions and 0 deletions
@@ -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);