From fd077cd54059106a8ce3ab2214028e812b617c16 Mon Sep 17 00:00:00 2001
From: Mikael Huppunen <mikael.huppunen@gmail.com>
Date: Thu, 25 Jun 2026 16:18:19 +0300
Subject: [PATCH] Created a function for getting pool size

---
 mpi_pancake.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mpi_pancake.cpp b/mpi_pancake.cpp
index 2e7217f..aa83edc 100644
--- a/mpi_pancake.cpp
+++ b/mpi_pancake.cpp
@@ -331,6 +331,9 @@ static bool initialized = false;
 static gpuStream_t s = nullptr;
 static std::unordered_map<MPI_Request, Pending *> pending;
 
+extern "C" std::size_t get_pool_size() {
+  return POOL;
+}
 
 __global__ void pack_kernel(const char *__restrict__ src,
                             const int64_t *__restrict__ disp,