From dfcba01db602b8481535b9baad7cd8caa803fd18 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
---
 samples/2_Cookbook/8_peer2peer/peer2peer.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/samples/2_Cookbook/8_peer2peer/peer2peer.cpp b/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
index 990599e1cb..0f532a2f0a 100644
--- a/samples/2_Cookbook/8_peer2peer/peer2peer.cpp
+++ b/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);