Merge pull request #1072 from kpyzhov/master

Refined hipSetDevice test. [ROCm/hip commit: fea21dc6d5]
2019-05-07 16:07:36 +05:30
parent b6393eb059 45eb04ea5c
commit d6e2d71d99
1 changed files with 16 additions and 6 deletions
@@ -1,9 +1,19 @@
-#include "hip/hip_runtime.h"
-#include "hip/hip_runtime_api.h"
-#include "hipDeviceUtil.h"
+#include "test_common.h"

 int main() {
-    HIP_CHECK(hipSetDevice(0), hipSetDevice);
-    HIP_CHECK(hipSetDevice(1026), hipSetDevice);
-    HIP_CHECK(hipSetDevice(-1), hipSetDevice);
+    int numDevices = 0;
+
+    HIPCHECK_API(hipGetDeviceCount(&numDevices), hipSuccess);
+    if (numDevices > 0) {
+        for (int deviceId = 0; deviceId < numDevices; deviceId++) {
+            HIPCHECK_API(hipSetDevice(deviceId), hipSuccess);
+        }
+        HIPCHECK_API(hipSetDevice(numDevices), hipErrorInvalidDevice);
+        HIPCHECK_API(hipSetDevice(-1), hipErrorInvalidDevice);
+    }
+    else {
+        failed("Error: failed to find any compatible devices.");
+    }
+
+    passed();
 }