Merge branch 'memtracker' of https://github.com/AMDComputeLibraries/HIP-privatestaging into memtracker

hipamd/tests/src/hipMemcpy.cpp

@@ -223,6 +223,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() );

hipamd/tests/src/hipPointerAttrib.cpp

@@ -301,6 +301,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));
@@ -477,7 +478,6 @@ int main(int argc, char *argv[])
     N= 1000000;
     HipTest::parseStandardArguments(argc, argv, true);
 
-    HIPCHECK(hipSetDevice(p_gpuDevice));
 
     Nbytes = N*sizeof(char);
 
@@ -485,6 +485,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();
     }