cuFFTDx APIs (nvmath.device)#

Overview#

These APIs offer integration with the NVIDIA cuFFTDx library.Detailed documentation of cuFFTDx can be found in thecuFFTDx documentation.

Note

TheFFT device APIs in modulenvmath.device currently support cuFFTDx 1.5.1, also availableas part of MathDx 25.06. All functionalities from the C++ library are supported withthe exception of cuFFTDx C++ APIs with a workspace argument, which are currently notavailable in nvmath-python.

API Reference#

fft(*[, compiler, code_type, execute_api])

Create anFFT object that encapsulates a compiled and ready-to-use FFT device function.

FFT(size, precision, fft_type, execution, *)

A class that encapsulates a partial FFT device function.