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

Fix combine_function

Browse Source
This commit is contained in:
DoHoonKim8
2024-11-14 08:42:38 +00:00
parent ddd34a7985
commit 2ee7f87640
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,7 +12,7 @@ __device__ fr combine_function(fr* evals, unsigned int start, unsigned int strid
unsigned int table_dim = num_args - 2;
fr result = fr::zero();
for (int i = 1; i <= table_dim; i++)
result += fr(1 << (num_args - 2 - i)) * evals[start + i * stride];
result += fr(1 << (i - 1)) * evals[start + i * stride];
result = evals[start] - result;
for (int i = 1; i <= table_dim; i++)
result += gamma->pow(i) * evals[start + i * stride] * (evals[start + i * stride] - fr::one());