Fix hipStreamWaitEvent for single GPU.

[ROCm/clr commit: ae9fdf9bc1]
2017-05-10 13:23:49 -05:00
parent b5f0a9471d
commit 501d0b3161
1 changed files with 1 additions and 1 deletions
@@ -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++) {