Skip to content
Snippets Groups Projects
Commit 93975db2 authored by Maximillian Bensch's avatar Maximillian Bensch
Browse files

clFFT_GetSize: add addtional allocs from exectute step

parent ef597f83
No related branches found
No related tags found
No related merge requests found
......@@ -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;
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment