From 39a63b7d32ecac0920b42b495417817f7c10ef70 Mon Sep 17 00:00:00 2001
From: Tom Deakin <tom.deakin@bristol.ac.uk>
Date: Thu, 16 Jul 2015 18:14:11 +0100
Subject: [PATCH] CUDA fix bug with listing device names

---
 cuda-stream.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cuda-stream.cu b/cuda-stream.cu
index 19304b8..6ac7453 100644
--- a/cuda-stream.cu
+++ b/cuda-stream.cu
@@ -321,6 +321,7 @@ void parseArguments(int argc, char *argv[])
                 std::cout << "Devices:" << std::endl;
                 for (int i = 0; i < count; i++)
                 {
+                    cudaSetDevice(i);
                     std::cout << i << ": " << getDeviceName() << std::endl;
                 }
                 std::cout << std::endl;