cuFFTDx API (nvmath.device)

概述

这些 API 提供与 NVIDIA cuFFTDx 库的集成。cuBLASDx 的详细文档可以在 cuFFTDx 文档 中找到。

注意

模块 nvmath.device 中的 fft 设备 API 当前支持 cuFFTDx 1.2.0，也作为 MathDx 24.04 的一部分提供。C++ 库的所有功能都受到支持，但带有工作区参数的 cuFFTDx C++ API 除外，这些 API 目前在 nvmath-python 中不可用。

API 参考

fft(*[, compiler])	创建一个封装已编译且可直接使用的 FFT 设备函数的 FFTOptions 对象。
FFTOptions(size, precision, fft_type, ...[, ...])	一个封装部分 FFT 设备函数的类。