From 8a9b8a19bef2f5b09946fef01046287dc83a6685 Mon Sep 17 00:00:00 2001
From: Maximillian Bensch <maximillian.bensch@aei.mpg.de>
Date: Fri, 26 Feb 2021 09:07:10 +0100
Subject: [PATCH] clFFT_GetSize: add batchSize argument
---
include/clFFT.h | 2 +-
src/fft_setup.cpp | 5 ++---
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/include/clFFT.h b/include/clFFT.h
index af75da8..5eb803d 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 47cb838..ce14116 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;
}
--
GitLab