Fix tests to account for multi-gpu

2016-02-15 05:19:52 -06:00
parent 8939b4f0e5
commit afbe451b0d
2 changed files with 6 additions and 1 deletions
@@ -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() );