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

fix: NTT release domain linkage

Browse Source
This commit is contained in:
Yuval Shekel
2024-04-03 13:09:13 +00:00
committed by yshekel
parent 25ac705c3b
commit 406020bda6
4 changed files with 46 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/c++/polynomial_multiplication/example.cu
View File

@@ -84,7 +84,7 @@ int main(int argc, char** argv)
// (4) multiply A,B
CHK_IF_RETURN(cudaMallocAsync(&MulGpu, sizeof(test_data) * NTT_SIZE, ntt_config.ctx.stream));
vec_ops::VecOpsConfig<test_data> config {
vec_ops::VecOpsConfig<test_data> config{
ntt_config.ctx,
true, // is_a_on_device
true, // is_b_on_device
@@ -92,8 +92,7 @@ int main(int argc, char** argv)
false, // is_montgomery
false // is_async
};
CHK_IF_RETURN(
vec_ops::Mul(GpuA, GpuB, NTT_SIZE, config, MulGpu));
CHK_IF_RETURN(vec_ops::Mul(GpuA, GpuB, NTT_SIZE, config, MulGpu));
// (5) INTT (in place)
ntt_config.are_inputs_on_device = true;
@@ -118,6 +117,7 @@ int main(int argc, char** argv)
benchmark(false); // warmup
benchmark(true, 20);
ntt::ReleaseDomain<test_scalar>(ntt_config.ctx);
CHK_IF_RETURN(cudaStreamSynchronize(ntt_config.ctx.stream));
return 0;