A Python wrapper for the OpenCL FFT library clFFT from AMD

AMD has created a open source FFT library for GPU devices based on OpenCL, called clFFT (released under Apache 2.0 license).

This python wrapper is designed to tightly integrate with pyopencl. It consists of a low-level cython based wrapper with an interface similar to the underlying C library. On top of that it offers a easy-to-use high-level interface designed to work on data contained in instances of pyopencl.array.Array, a numpy work-alike array class. The high-level interface is similar to that of pyFFTW, a python wrapper for the FFTW library. For details of the high-level interface see fft.py.

Compared to pyfft, a python implementation of Apple's FFT library, AMD's FFT library offers some additional features such as transform sizes that are powers of 2,3 and 5, and real-to-complex transforms. And on AMD hardware a better performance can be expected, e.g., gpyfft: 280 Gflops compared to pyfft: 63 GFlops (for single precision, accurate math, inplace, transform size 1024x1024, batch size 4, on AMD Cayman, HD6950).

This wrapper is functional, the high-level interface is not completely settled.

• low level wrapper (mostly) completed
• high level wrapper: complex (single precision), interleaved data, in and out of place (some tests and benchmarking available)
• creation of pyopencl Events for synchronization

• debug mode to output generated kernels
• documentation for low level wrapper (instead refer to library doc)
• define API for high level interface
• high-level interface: double precision data, planar data, real<->complex transforms
• high-level interface: tests for non-contiguous data
• handling of batched transforms in the general case, e.g. shape (4,5,6), axes = (1,), i.e., more than one axes where no transform is performed. (not always possible with single call for arbitrary strides, need to figure out when possible)
• high-level interface: implement some strategy to deliver optimal performance (e.g. order of transforms along axes for 2D, 3D transforms depending on memory layout)

• python (tested with 2.7, should work with 3.x)
• pyopencl (>v2013.2 required)
• distribute
• cython
• AMD clFFT

1. Build and install the AMD clFFT library:
   • install clFFT
   • edit setup.py to point to clFFT directory

Then, either:

2. python setup.py install

or for developing

3. python setup.py develop

   python setup.py build_ext --inplace (rerun last command whenever cython extension has changed)

   python setup.py develop -u #for uninstalling development version

Requirements:

• C/C++ Compiler. Tested with free compilers (64bit) from Microsoft Windows SDK v7.0
• cmake (3.0)
• OpenCL environment (tested with AMD APP SDK, 2.9)

1. Download clFFT from github

   git checkout https://github.com/clMathLibraries/clFFT.git

2) in .../clFFT/src, open SDK command shell (Start - Programs - Microsoft Windows SDK v7.1 - CMD Shell)

setenv /Release cmake -G "NMake Makefile" nmake

or use cmake-gui, with source code .../clFFT/src, build dir .../clFFT/src, manually change CMAKE/CMAKE_BUILD_TYPE to Release

In .../clFFT/src/staging should contain clFFT.dll.

3) in gpyfft/setup.py check that in setup.py CLFFT_DIR points to the clFFT folder, and CL_INCL_DIRS to the OpenCL headers. Note that the setup script copies the clFFT binary libs (clFFT.dll, ...) to the package directory.

For some basic testing, run in the base directory of this wrapper:

python gpyfft/test_simple.py

or for some benchmarking:

python gpyfft/fft.py

LGPL

This wrapper has been tested on Windows 7 (64bit) with AMD Radeon 6950, and OS X 10.7, 10.9 with Nvidia GT330M. Success reports for more recent systems are welcome!

3. Gregor Thalhammer 2015