1 files changed, 4 insertions, 4 deletions

diff --git a/constant_cycles_kernel.cu b/constant_cycles_kernel.cu
index b45ae8c..6748552 100644
--- a/constant_cycles_kernel.cu
+++ b/constant_cycles_kernel.cu
@@ -8,8 +8,8 @@
 
 #include "testbench.h"
 
-__global__ void loop_on_gpu(unsigned long iters, int *__unused) {
-        for (volatile int i = 0; i < iters; i++) (*__unused)++;
+__global__ void loop_on_gpu(uint64_t iters, int *__unused) {
+        for (volatile uint32_t i = 0; i < iters; i++) (*__unused)++;
 }
 
 int main(int argc, char **argv) {
@@ -23,8 +23,8 @@ int main(int argc, char **argv) {
         }
 
         // Input is multiplied by one million, unless infinite
-        unsigned long num_iters = strtoul(argv[1], NULL, 10);
-        if (num_iters != (unsigned long)(-1))
+        uint64_t num_iters = strtoull(argv[1], NULL, 10);
+        if (num_iters != (uint64_t)(-1))
                 num_iters *= 1000 * 1000;
 
         // Initialize CUDA and a context (hack)