fix overflow

okx · Sep 6, 2024 · cea27ba · cea27ba
1 parent 7399eb9
commit cea27ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/prover/prover.cpp b/src/prover/prover.cpp
@@ -115,7 +115,7 @@ Prover::Prover(Goldilocks &fr,
 
 #if defined(__USE_CUDA__) && defined(ENABLE_EXPERIMENTAL_CODE)
             warmup_gpu();
-            alloc_pinned_mem_per_device((1 << _starkInfo.starkStruct.nBitsExt) * 32);
+            alloc_pinned_mem_per_device((uint64_t(1) << _starkInfo.starkStruct.nBitsExt) * 32);
 #endif
 
             // Allocate an area of memory, mapped to file, to store all the committed polynomials,