From 00de932454a27e36e479a60e1d3a979fe84178ca Mon Sep 17 00:00:00 2001 From: Tom Deakin Date: Tue, 12 Jan 2021 10:09:55 +0000 Subject: [PATCH] Save array size argument as signed integer --- main.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/main.cpp b/main.cpp index 2816009..cd1c8df 100644 --- a/main.cpp +++ b/main.cpp @@ -44,7 +44,7 @@ #endif // Default size of 2^25 -unsigned int ARRAY_SIZE = 33554432; +int ARRAY_SIZE = 33554432; unsigned int num_times = 100; unsigned int deviceIndex = 0; bool use_float = false; @@ -538,11 +538,13 @@ void parseArguments(int argc, char *argv[]) else if (!std::string("--arraysize").compare(argv[i]) || !std::string("-s").compare(argv[i])) { - if (++i >= argc || !parseUInt(argv[i], &ARRAY_SIZE)) + unsigned int tmp_array_size = static_cast(ARRAY_SIZE); + if (++i >= argc || !parseUInt(argv[i], &tmp_array_size)) { std::cerr << "Invalid array size." << std::endl; exit(EXIT_FAILURE); } + ARRAY_SIZE = static_cast(tmp_array_size); } else if (!std::string("--numtimes").compare(argv[i]) || !std::string("-n").compare(argv[i]))