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