update

okx · Sep 3, 2024 · 12446c6 · 12446c6
1 parent afc6c89
commit 12446c6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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);