cory-master/BabelStream
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add parseInt to cuda version for device index

Browse Source
This commit is contained in:
Tom Deakin 2015-07-16 17:33:58 +01:00
parent 8c7224ab42
commit 3ff57ba9d6
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cuda-stream.cu
View File

@ -286,6 +286,13 @@ int parseUInt(const char *str, unsigned int *output)
return !strlen(next); 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[]) void parseArguments(int argc, char *argv[])
{ {
for (int i = 1; i < argc; i++) for (int i = 1; i < argc; i++)
@ -315,7 +322,7 @@ void parseArguments(int argc, char *argv[])
} }
else if (!strcmp(argv[i], "--device")) 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; std::cout << "Invalid device index" << std::endl;
exit(1); exit(1);