clFFT_GetSize: add batchSize argument

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
 }

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;
   }