cuFFTDx APIs (nvmath.device)

Overview

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

Note

The FFT device APIs in module nvmath.device currently support cuFFTDx 1.6.1, also available as part of MathDx 25.12.1. 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, code_type, execute_api])	Create an FFT 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.