From 746549b28d3b654e0670feca0065f51affdb7db8 Mon Sep 17 00:00:00 2001
From: Stefano Salsano <stefano.salsano@uniroma2.it>
Date: Fri, 14 Jun 2024 11:28:55 +0200
Subject: [PATCH] improve parsing of stepbytes (increment size) argument

---
 src/common.cu | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/common.cu b/src/common.cu
index 04e81422f0..7706dd9a54 100644
--- a/src/common.cu
+++ b/src/common.cu
@@ -764,7 +764,12 @@ int main(int argc, char* argv[]) {
         maxBytes = (size_t)parsed;
         break;
       case 'i':
-        stepBytes = strtol(optarg, NULL, 0);
+        parsed = parsesize(optarg);
+        if (parsed < 0) {
+          fprintf(stderr, "invalid size specified for 'stepBytes'\n");
+          return -1;
+        }
+        stepBytes = (size_t)parsed;
         break;
       case 'f':
         stepFactor = strtol(optarg, NULL, 0);