diff --git a/include/clFFT.h b/include/clFFT.h index af75da88317d85b988cb0b4dad8dd004078aa4e9..5eb803d2275bed1a4d0e664b0fb35a02fcf5d370 100644 --- a/include/clFFT.h +++ b/include/clFFT.h @@ -194,7 +194,7 @@ cl_int clFFT_1DTwistPlannar(clFFT_Plan Plan, cl_command_queue queue, cl_mem arra void clFFT_DumpPlan( clFFT_Plan plan, FILE *file); -cl_int clFFT_GetSize(clFFT_Plan Plan, size_t* workSize); +cl_int clFFT_GetSize(clFFT_Plan Plan, size_t* workSize, cl_uint batchSize); #ifdef __cplusplus } diff --git a/src/fft_setup.cpp b/src/fft_setup.cpp index 47cb838b78e15daf7aee884361b970f9a2140e49..ce14116d047ab5e084ec62445b7245fa2359d4da 100644 --- a/src/fft_setup.cpp +++ b/src/fft_setup.cpp @@ -528,7 +528,7 @@ void clFFT_DumpPlan( clFFT_Plan Plan, FILE *file) } cl_int -clFFT_GetSize(clFFT_Plan Plan, size_t* workSize) +clFFT_GetSize(clFFT_Plan Plan, size_t* workSize, cl_uint batchSize) { if ( workSize == NULL ) { return CL_INVALID_VALUE; @@ -538,9 +538,8 @@ clFFT_GetSize(clFFT_Plan Plan, size_t* workSize) if (plan->twiddleMethod == clFFT_TaylorLUT || plan->twiddleMethod == clFFT_BigLUT) { *workSize += ( plan->N1 + plan->N2 ) * 2 * sizeof(float); } - // Additional alloc from allocateTemporaryBufferInterleaved() or allocateTemporaryBufferPlannar() + // Additional alloc's 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; }