Update reservation policies

alpaka-group · Nov 8, 2024 · cc3c172 · cc3c172
1 parent 99bcf21
commit cc3c172
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/include/mallocMC/reservePoolPolicies/AlpakaBuf.hpp b/src/include/mallocMC/reservePoolPolicies/AlpakaBuf.hpp
@@ -45,7 +45,7 @@ namespace mallocMC
                 return alpaka::getPtrNative(*poolBuffer);
             }
 
-            void resetMemPool()
+            void resetMemPool(void* p)
             {
                 poolBuffer = {};
             }

diff --git a/src/include/mallocMC/reservePoolPolicies/CudaSetLimits.hpp b/src/include/mallocMC/reservePoolPolicies/CudaSetLimits.hpp
@@ -63,7 +63,7 @@ namespace mallocMC
                 return nullptr;
             }
 
-            static void resetMemPool()
+            static void resetMemPool(void* p = nullptr)
             {
                 cudaDeviceSetLimit(cudaLimitMallocHeapSize, 8192U);
                 cudaGetLastError(); // cudaDeviceSetLimit() usually fails if any