StarPU Handbook - StarPU Extensions
Loading...
Searching...
No Matches
16. FFT Support

StarPU provides libstarpufft, a library whose design is very similar to both fftw and cufft, the difference being that it takes benefit from both CPUs and GPUs. It should however be noted that GPUs do not have the same precision as CPUs, so the results may be different by a negligible amount.

Different precisions are available, namely float, double and long double precisions, with the following fftw naming conventions:

  • double precision structures and functions are named e.g. starpufft_execute()
  • float precision structures and functions are named e.g. starpufftf_execute()
  • long double precision structures and functions are named e.g. starpufftl_execute()

The documentation below is given with names for double precision, replace starpufft_ with starpufftf_ or starpufftl_ as appropriate.

Only complex numbers are supported at the moment.

The application has to call starpu_init() before calling starpufft functions.

Either main memory pointers or data handles can be provided.

  • To provide main memory pointers, use starpufft_start() or starpufft_execute(). Only one FFT can be performed at a time, because StarPU will have to register the data on the fly. In the starpufft_start() case, starpufft_cleanup() needs to be called to unregister the data.
  • To provide data handles (which is preferable), use starpufft_start_handle() (preferred) or starpufft_execute_handle(). Several FFTs tasks can be submitted for a given plan, which permits e.g. to start a series of FFT with just one plan. starpufft_start_handle() is preferable since it does not wait for the task completion, and thus permits to enqueue a series of tasks.

All functions are defined in FFT Support.

Some examples illustrating the usage of FFT API are available in the directory starpufft/tests.

16.1 Compilation

The flags required to compile or link against the FFT library are accessible with the following commands:

$ pkg-config --cflags starpufft-1.4  # options for the compiler
$ pkg-config --libs starpufft-1.4    # options for the linker

Also pass the option –static if the application is to be linked statically.