def __init__(self,nx,ny,bsize=16):
self.nx = np.int32(nx)
self.ny = np.int32(ny)
self.bsize = bsize
self.batch = nx/bsize*ny
from pycudafft import FFTPlan
self.fftplan = FFTPlan(self.bsize,1)
def __init__(self,nx,ny,bsize=16):
self.nx = np.int32(nx)
self.ny = np.int32(ny)
self.bsize = bsize #block size
self.batch = nx/bsize*ny #How many 1D-FFTs?
self.fftplan = FFTPlan(self.bsize,1)
