update

okx · Sep 3, 2024 · afc6c89 · afc6c89
1 parent ffb1790
commit afc6c89
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/starkpil/chelpers_steps_gpu.cu b/src/starkpil/chelpers_steps_gpu.cu
@@ -198,6 +198,7 @@ void CHelpersStepsGPU::calculateExpressionsRowsGPU(StarkInfo &starkInfo, StepsPa
 
     if ((rowEnd - rowIni) < nrowsPack*nCudaThreads*nStreams*nDevices) {
         nCudaThreads = (rowEnd - rowIni) / (nrowsPack*nStreams*nDevices);
+        subDomainSize = nrowsPack * nCudaThreads;
         printf("nCudaThreads: %u\n", nCudaThreads);
     }