wgsl-fft 0.3.1

GPU-accelerated FFT using Webgpu compute shaders
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

//! GPU-accelerated FFT using [wgpu](https://github.com/gfx-rs/wgpu) compute shaders.

pub mod benchmark;
#[cfg(feature = "cuda")]
mod cufft_wrapper;
#[cfg(feature = "hipfft")]
pub mod hipfft_wrapper;
pub mod rivals;
#[cfg(feature = "rocm")]
mod rocfft_wrapper;
pub mod shaders;

mod fft;
mod pipelines;

pub use fft::{FftExecutor, FftUniforms, GpuFft, GpuFftTrait, SizeCache};
pub use pipelines::{FftDirection, FftPipelines};

#[cfg(feature = "cuda")]
pub use cufft_wrapper::CuFft;
#[cfg(feature = "hipfft")]
pub use hipfft_wrapper::HipFft;
#[cfg(feature = "rocm")]
pub use rocfft_wrapper::RocFft;