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

---
 src/starkpil/chelpers_steps_gpu.cu | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/starkpil/chelpers_steps_gpu.cu b/src/starkpil/chelpers_steps_gpu.cu
index e4633e1c9..6f7752ed3 100644
--- 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;