From 7e46e9059184b8d826e2c745d12d0bebbdf7a5e9 Mon Sep 17 00:00:00 2001
From: Ben Sander <ben.sander@amd.com>
Date: Mon, 15 Feb 2016 05:19:52 -0600
Subject: [PATCH] Fix tests to account for multi-gpu

[ROCm/clr commit: afbe451b0da2a25b1e5f05497f04e6286e45f12c]
---
 projects/clr/hipamd/tests/src/hipMemcpy.cpp        | 3 +++
 projects/clr/hipamd/tests/src/hipPointerAttrib.cpp | 4 +++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/tests/src/hipMemcpy.cpp b/projects/clr/hipamd/tests/src/hipMemcpy.cpp
index 3502b81e9d..b76f98c687 100644
--- a/projects/clr/hipamd/tests/src/hipMemcpy.cpp
+++ b/projects/clr/hipamd/tests/src/hipMemcpy.cpp
@@ -212,6 +212,9 @@ int main(int argc, char *argv[])
 {
     HipTest::parseStandardArguments(argc, argv, true);
 
+    printf ("info: set device to %d\n", p_gpuDevice);
+    HIPCHECK(hipSetDevice(p_gpuDevice));
+
 
     if (p_tests & 0x1) {
         HIPCHECK ( hipDeviceReset() );
diff --git a/projects/clr/hipamd/tests/src/hipPointerAttrib.cpp b/projects/clr/hipamd/tests/src/hipPointerAttrib.cpp
index 1418997274..12856da984 100644
--- a/projects/clr/hipamd/tests/src/hipPointerAttrib.cpp
+++ b/projects/clr/hipamd/tests/src/hipPointerAttrib.cpp
@@ -289,6 +289,7 @@ void clusterAllocs(int numAllocs, size_t minSize, size_t maxSize)
 
     for (int i =0; i<numDevices; i++) {
         size_t free, total;
+        HIPCHECK(hipSetDevice(i));
         HIPCHECK(hipMemGetInfo(&free, &total));
         printf ("  device#%d: hipMemGetInfo: free=%zu (%4.2fMB) clusterAllocTotalDevice=%lu (%4.2fMB) total=%zu (%4.2fMB)\n", 
                 i, free, (float)(free/1024.0/1024.0), totalDeviceAllocated[i], (float)(totalDeviceAllocated[i])/1024.0/1024.0, total, (float)(total/1024.0/1024.0));
@@ -452,7 +453,6 @@ int main(int argc, char *argv[])
     N= 1000000;
     HipTest::parseStandardArguments(argc, argv, true);
 
-    HIPCHECK(hipSetDevice(p_gpuDevice));
 
     Nbytes = N*sizeof(char);
 
@@ -460,6 +460,8 @@ int main(int argc, char *argv[])
 
 
     if (p_tests & 0x01) {
+        printf ("info: set device to %d\n", p_gpuDevice);
+        HIPCHECK(hipSetDevice(p_gpuDevice));
         testSimple();
     }