From 93975db29bb45a494fb3501ca8ce4907e34c4b7f Mon Sep 17 00:00:00 2001
From: Maximillian Bensch <maximillian.bensch@aei.mpg.de>
Date: Thu, 25 Feb 2021 14:52:48 +0100
Subject: [PATCH] clFFT_GetSize: add addtional allocs from exectute step

---
 src/fft_setup.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/fft_setup.cpp b/src/fft_setup.cpp
index 2914a3b..4944043 100644
--- a/src/fft_setup.cpp
+++ b/src/fft_setup.cpp
@@ -542,6 +542,12 @@ clFFT_GetSize(clFFT_Plan Plan, size_t* workSize)
   }
   cl_fft_plan *plan = (cl_fft_plan *) Plan;
   *workSize = plan->tempmemobj.size + plan->tempmemobj_real.size + plan->tempmemobj_imag.size + plan->cossin_LUT_d1.size;
+  // Additional alloc from allocateTemporaryBufferInterleaved() or allocateTemporaryBufferPlannar()
+  if(plan->temp_buffer_needed) {
+      cl_uint batchSize = 1;
+      size_t tmpLength = plan->n.x * plan->n.y * plan->n.z * batchSize * 2 * sizeof(cl_float);
+      *workSize += tmpLength;
+  }
 
   return CL_SUCCESS;
 }
-- 
GitLab