cuFFTDx APIs (nvmath.device)#

Overview#

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

Note

The fft device APIs in module nvmath.device currently support cuFFTDx 1.2.0, also available as part of MathDx 24.04. All functionalities from the C++ library are supported with the exception of cuFFTDx C++ APIs with a workspace argument, which are currently not available in nvmath-python.

API Reference#

fft(*[, compiler])

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

FFTOptions(size, precision, fft_type, ...[, ...])

A class that encapsulates a partial FFT device function.