From e6c4873927faa0ba1e04b166df5bb10e9c532f31 Mon Sep 17 00:00:00 2001
From: xzavier <yann.sjtu@gmail.com>
Date: Tue, 3 Sep 2024 16:39:22 +0800
Subject: [PATCH] update

---
 src/starkpil/chelpers_steps_gpu.cu | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/starkpil/chelpers_steps_gpu.cu b/src/starkpil/chelpers_steps_gpu.cu
index 886ef6c6b..afe84c426 100644
--- a/src/starkpil/chelpers_steps_gpu.cu
+++ b/src/starkpil/chelpers_steps_gpu.cu
@@ -20,12 +20,11 @@ cudaStream_t streams[nStreams*MAX_GPUS];
 
 void CHelpersStepsGPU::prepareGPU(StarkInfo &starkInfo, StepsParams &params, ParserArgs &parserArgs, ParserParams &parserParams) {
 
+    nrowsPack = 8;
     prepare(starkInfo, params, parserArgs, parserParams);
 
-
     CHECKCUDAERR(cudaGetDeviceCount(&nDevices));
     printf("nDevices: %d\n", nDevices);
-    nrowsPack = 8;
     nCudaThreads = 1<<14;
     domainExtended = parserParams.stage > 3 ? true : false;
     domainSize = domainExtended ? 1 << starkInfo.starkStruct.nBitsExt : 1 << starkInfo.starkStruct.nBits;