From 3ff57ba9d6093822686fe42e531f23cb2b3969ad Mon Sep 17 00:00:00 2001
From: Tom Deakin <tom.deakin@bristol.ac.uk>
Date: Thu, 16 Jul 2015 17:33:58 +0100
Subject: [PATCH] Add parseInt to cuda version for device index

---
 cuda-stream.cu | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/cuda-stream.cu b/cuda-stream.cu
index 4a2c114..665b416 100644
--- a/cuda-stream.cu
+++ b/cuda-stream.cu
@@ -286,6 +286,13 @@ int parseUInt(const char *str, unsigned int *output)
     return !strlen(next);
 }
 
+int parseInt(const char *str, int *output)
+{
+    char *next;
+    *output = strtol(str, &next, 10);
+    return !strlen(next);
+}
+
 void parseArguments(int argc, char *argv[])
 {
     for (int i = 1; i < argc; i++)
@@ -315,7 +322,7 @@ void parseArguments(int argc, char *argv[])
         }
         else if (!strcmp(argv[i], "--device"))
         {
-            if (++i >= argc || !parseUInt(argv[i], &deviceIndex))
+            if (++i >= argc || !parseInt(argv[i], &deviceIndex))
             {
                 std::cout << "Invalid device index" << std::endl;
                 exit(1);