def _get_3dplan(shape):
debug("[*] Creating a 3D FFT plan...")
plan = cufft.cufftHandle()
cufft.cufftPlan3d(plan, shape[0], shape[1], shape[2], cufft.CUFFT_C2C)
return plan
def _get_1dplan(shape,batch=1):
debug("[*] Creating a 1D FFT plan...")
plan = cufft.cufftHandle()
cufft.cufftPlan1d(plan, shape[0], cufft.CUFFT_C2C, batch)
return plan