Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
yann-sjtu committed Sep 3, 2024
1 parent 5a3a4e5 commit e6c4873
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/starkpil/chelpers_steps_gpu.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit e6c4873

Please sign in to comment.