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,