pseXperiments/cuda-sumcheck
1
0
Fork 0
mirror of https://github.com/pseXperiments/cuda-sumcheck.git synced 2026-01-09 20:37:55 -05:00
Code Issues Packages Projects Releases 1 Wiki Activity

Avoid unnecessary computation

Browse Source
This commit is contained in:
DoHoonKim8
2024-08-30 08:02:32 +00:00
committed by DoHoon Kim
parent fb94e9e80f
commit 58d1b497ba
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sumcheck/src/gpu/cuda/kernels/sumcheck.cu
View File

@@ -44,7 +44,9 @@ extern "C" __global__ void fold_into_half(
const int buf_offset = (blockIdx.x / num_blocks_per_poly) * stride;
const int poly_offset = (blockIdx.x / num_blocks_per_poly) * initial_poly_size;
while (tid < stride) {
buf[buf_offset + tid] = (*challenge) * (polys[poly_offset + tid + stride] - polys[poly_offset + tid]) + polys[poly_offset + tid];
if (*challenge == fr::zero()) buf[buf_offset + tid] = polys[poly_offset + tid];
else if (*challenge == fr::one()) buf[buf_offset + tid] = polys[poly_offset + tid + stride];
else buf[buf_offset + tid] = (*challenge) * (polys[poly_offset + tid + stride] - polys[poly_offset + tid]) + polys[poly_offset + tid];
tid += blockDim.x * num_blocks_per_poly;
}
}