Python slab_FFT Examples

Example #1

File: test_FFT.py Project: bkmgit/mpiFFT4py

def FFT(request):
    prec = {"s": "single", "d": "double"}[request.param[-1]]
    if request.param[:3] == "pen":
        return pencil_FFT(array([N, N, N]),
                          L,
                          MPI,
                          prec,
                          None,
                          alignment=string.upper(request.param[-2]))
    else:
        return slab_FFT(array([N, N, N]), L, MPI, prec)

Example #2

File: test_FFT.py Project: t-bltg/mpiFFT4py

def FFT_padded(request):
    prec = {"s": "single", "d": "double"}[request.param[-1]]
    if request.param[:3] == "pen":
        if request.param[-3] == 's':
            return pencil_FFT(array([N, N, N]),
                              L,
                              MPI,
                              prec,
                              None,
                              method='Swap',
                              alignment=string.upper(request.param[-2]))
        elif request.param[-3] == 'n':
            return pencil_FFT(array([N, N, N]),
                              L,
                              MPI,
                              prec,
                              None,
                              method='Nyquist',
                              alignment=string.upper(request.param[-2]))
    else:
        return slab_FFT(array([N, N, N]), L, MPI, prec)

Example #3

File: test_FFT.py Project: bkmgit/mpiFFT4py

def FFT_padded(request):
    prec = {"s": "single", "d": "double"}[request.param[-1]]
    return slab_FFT(array([N, N, N]), L, MPI, prec)