update

okx · Sep 3, 2024 · f802d69 · f802d69
1 parent e6c4873
commit f802d69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/starkpil/chelpers_steps_gpu.cu b/src/starkpil/chelpers_steps_gpu.cu
@@ -20,12 +20,12 @@ cudaStream_t streams[nStreams*MAX_GPUS];
 
 void CHelpersStepsGPU::prepareGPU(StarkInfo &starkInfo, StepsParams &params, ParserArgs &parserArgs, ParserParams &parserParams) {
 
-    nrowsPack = 8;
+    nrowsPack = 1;
     prepare(starkInfo, params, parserArgs, parserParams);
 
     CHECKCUDAERR(cudaGetDeviceCount(&nDevices));
     printf("nDevices: %d\n", nDevices);
-    nCudaThreads = 1<<14;
+    nCudaThreads = 1<<17;
     domainExtended = parserParams.stage > 3 ? true : false;
     domainSize = domainExtended ? 1 << starkInfo.starkStruct.nBitsExt : 1 << starkInfo.starkStruct.nBits;
     subDomainSize = nrowsPack * nCudaThreads;