From ae9fdf9bc10835aeeed662ae74da6f39548e926d Mon Sep 17 00:00:00 2001
From: Ben Sander <ben.sander@amd.com>
Date: Wed, 10 May 2017 13:23:49 -0500
Subject: [PATCH] Fix hipStreamWaitEvent for single GPU.

---
 hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp b/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
index 1d9ec45685..adf0d4af0c 100644
--- a/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
+++ b/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
@@ -395,7 +395,7 @@ int main(int argc, char *argv[])
 
     int numDevices;
     HIPCHECK(hipGetDeviceCount(&numDevices));
-    numDevices =2; // TODO - remove me.
+    numDevices = min(2, numDevices); // multi-GPU to 2 device.
 
     for (int d=0; d<numDevices/*TODO*/; d++) {
         for (int i=0; i<p_streams; i++) {