From a28399f456c9df8a6e84e246cff8302b5b142cb9 Mon Sep 17 00:00:00 2001
From: Sandeep Kumar <Sandeep.Kumar@amd.com>
Date: Thu, 4 May 2017 13:57:01 +0530
Subject: [PATCH] Print msg for single gpu

Change-Id: I2d23c73542add8973990ba96592016726994422e


[ROCm/hip commit: dfcba01db602b8481535b9baad7cd8caa803fd18]
---
 .../hip/samples/2_Cookbook/8_peer2peer/peer2peer.cpp   | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/projects/hip/samples/2_Cookbook/8_peer2peer/peer2peer.cpp b/projects/hip/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
index 990599e1cb..0f532a2f0a 100644
--- a/projects/hip/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
+++ b/projects/hip/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
@@ -55,13 +55,9 @@ void checkPeer2PeerSupport()
 {
     int gpuCount;
     int canAccessPeer;
-    int p2pCapableDeviceCount=0;
 
     HIPCHECK(hipGetDeviceCount(&gpuCount));
 
-    if (gpuCount < 2)
-        printf("Peer2Peer application requires atleast 2 gpu devices");
-
     for (int currentGpu=0; currentGpu<gpuCount; currentGpu++)
     {
         HIPCHECK(hipSetDevice(currentGpu));
@@ -161,6 +157,12 @@ int main(){
 
     HIPCHECK(hipGetDeviceCount(&gpuCount));
 
+    if (gpuCount < 2)
+    {
+        printf("Peer2Peer application requires atleast 2 gpu devices");
+        return 0;
+    }
+
     currentGpu = 0;
     peerGpu = (currentGpu + 1);