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)