From 12446c6711ee291eae656f474e1ab568aaa4c6d0 Mon Sep 17 00:00:00 2001 From: xzavier Date: Tue, 3 Sep 2024 16:16:46 +0800 Subject: [PATCH] update --- test/examples/main.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/test/examples/main.cpp b/test/examples/main.cpp index 50b47cf51..08d9895f0 100644 --- a/test/examples/main.cpp +++ b/test/examples/main.cpp @@ -10,6 +10,7 @@ #include "AllSteps.hpp" #include "zklog.hpp" #include "exit_process.hpp" +#include "cuda_utils.hpp" int main() { @@ -37,6 +38,10 @@ int main() StarkInfo starkInfo(starkInfoFile, reduceMemory); +#if defined(__USE_CUDA__) && defined(ENABLE_EXPERIMENTAL_CODE) + alloc_pinned_mem_per_device((1 << starkInfo.starkStruct.nBitsExt) * 32); +#endif + uint64_t polBits = starkInfo.starkStruct.steps[starkInfo.starkStruct.steps.size() - 1].nBits; FRIProof fproof((1 << polBits), FIELD_EXTENSION, starkInfo.starkStruct.steps.size(), starkInfo.evMap.size(), starkInfo.nPublics);