update

okx · Sep 3, 2024 · ce5ec29 · ce5ec29
1 parent c091c83
commit ce5ec29
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/starkpil/chelpers_steps_gpu.cu b/src/starkpil/chelpers_steps_gpu.cu
@@ -196,6 +196,11 @@ void CHelpersStepsGPU::calculateExpressionsRowsGPU(StarkInfo &starkInfo, StepsPa
         exitProcess();
     }
 
+    if ((rowEnd - rowIni) < nrowsPack*nCudaThreads*nStreams*nDevices) {
+        nCudaThreads = (rowEnd - rowIni) / (nrowsPack*nStreams*nDevices);
+        printf("nCudaThreads: %lu\n", nCudaThreads);
+    }
+
     assert((rowEnd - rowIni) % (nrowsPack*nCudaThreads*nStreams*nDevices) == 0);
     uint64_t nrowPerStream = (rowEnd - rowIni) / nStreams /nDevices;