From 037ce74fc94cf55e7c5ed8fb2d81b7fc8988ba5b Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Wed, 20 Dec 2017 09:36:00 +0530
Subject: [PATCH] Return pass on single gpu in hipPeerToPeer_simple

---
 hipamd/tests/src/p2p/hipPeerToPeer_simple.cpp | 42 +++++++++----------
 1 file changed, 20 insertions(+), 22 deletions(-)

diff --git a/hipamd/tests/src/p2p/hipPeerToPeer_simple.cpp b/hipamd/tests/src/p2p/hipPeerToPeer_simple.cpp
index 32cab371b8..c279658b5a 100644
--- a/hipamd/tests/src/p2p/hipPeerToPeer_simple.cpp
+++ b/hipamd/tests/src/p2p/hipPeerToPeer_simple.cpp
@@ -397,32 +397,30 @@ int main(int argc, char *argv[])
     if (gpuCount < 2)
     {
         printf("P2P application requires atleast 2 gpu devices\n");
-        return 0;
-    }
+    } else {
+        if (p_tests & 0x100) {
+            testPeerHostToDevice(false/*useAsyncCopy*/);
+        }
+        testPeerHostToDevice(true/*useAsyncCopy*/);
 
-    if (p_tests & 0x100) {
-        testPeerHostToDevice(false/*useAsyncCopy*/);
-    }
-    testPeerHostToDevice(true/*useAsyncCopy*/);
+        if (p_tests & 0x1) {
+            enablePeerFirst(false/*useAsyncCopy*/);
+        }
 
-    if (p_tests & 0x1) {
-        enablePeerFirst(false/*useAsyncCopy*/);
-    }
+        if (p_tests & 0x2) {
+            allocMemoryFirst(false/*useAsyncCopy*/);
+        }
 
-    if (p_tests & 0x2) {
-        allocMemoryFirst(false/*useAsyncCopy*/);
-    }
+        if (p_tests & 0x4) {
+            simpleNegative();
+        }
 
-    if (p_tests & 0x4) {
-        simpleNegative();
+        if (p_tests & 0x8) {
+            enablePeerFirst(true/*useAsyncCopy*/);
+        }
+        if (p_tests & 0x10) {
+            allocMemoryFirst(true/*useAsyncCopy*/);
+        }
     }
-
-    if (p_tests & 0x8) {
-        enablePeerFirst(true/*useAsyncCopy*/);
-    }
-    if (p_tests & 0x10) {
-        allocMemoryFirst(true/*useAsyncCopy*/);
-    }
-
     passed();
 }