예제 #1
0
파일: fft.py 프로젝트: npinto/python-cuda
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
예제 #2
0
파일: fft.py 프로젝트: npinto/python-cuda
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