VkFFT-A Performant, Cross-Platform and Open-Source GPU FFT Library

Abstract

The Fast Fourier Transform is an essential algorithm of modern computational science. The highly parallel structure of the FFT allows for its efficient implementation on graphics processing units (GPUs), which are now widely used for general-purpose computing. This paper presents the VkFFT - an efficient GPU-accelerated multidimensional Fast Fourier Transform library for Vulkan/CUDA/HIP/OpenCL/Level Zero and Metal projects. VkFFT aims to provide the community with a cross-platform open-source alternative to vendor-specific solutions while achieving comparable or better performance. This paper also discusses the optimizations implemented in VkFFT and verifies its performance and precision on modern high-performance computing GPUs. VkFFT is released under an MIT license.

Keywords

• FFT
• GPU
• parallel computing
• Vulkan
• CUDA
• HIP