Exemple #1
0
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
Exemple #2
0
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