cartan-gpu 0.5.1

Portable GPU compute primitives for the cartan ecosystem: wgpu device/buffer/kernel abstractions plus VkFFT-backed FFT.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

//! cartan-gpu: portable GPU compute primitives for the cartan ecosystem.
//!
//! See the crate-level docs for the architecture overview. This crate is
//! pre-release; the public API will stabilize once cartan-em v0.1 ships.

pub mod buffer;
pub mod device;
pub mod error;
pub mod fft;
pub mod kernel;
#[cfg(feature = "vkfft")]
pub mod hal_vulkan;
#[cfg(feature = "cuda")]
pub mod cuda;
#[cfg(feature = "cufft")]
pub mod cuda_buffer;
#[cfg(feature = "cufft")]
pub mod cufft;

pub use buffer::GpuBuffer;
pub use device::Device;
pub use error::GpuError;
pub use fft::{Fft, FftDirection};
pub use kernel::Kernel;

/// Re-export of the underlying `wgpu` crate so downstream consumers
/// don't need a separate `wgpu` dependency just to construct
/// `BufferUsages` for [`GpuBuffer`].
pub use wgpu;
#[cfg(feature = "vkfft")]
pub use fft::VkFftBackend;
#[cfg(feature = "cuda")]
pub use cuda::CudaDevice;
#[cfg(feature = "cufft")]
pub use cuda_buffer::CudaBuffer;
#[cfg(feature = "cufft")]
pub use cufft::CuFftBackend;

#[cfg(any(feature = "vkfft", feature = "cufft"))]
pub mod uni;
#[cfg(any(feature = "vkfft", feature = "cufft"))]
pub use uni::{UniBuffer, UniFftBackend};

#[cfg(all(feature = "vkfft", feature = "cufft", target_os = "linux"))]
pub mod shared_memory;
#[cfg(all(feature = "vkfft", feature = "cufft", target_os = "linux"))]
pub use shared_memory::SharedMemory;

#[cfg(all(feature = "vkfft", feature = "cufft", target_os = "linux"))]
pub mod shared_fft;
#[cfg(all(feature = "vkfft", feature = "cufft", target_os = "linux"))]
pub use shared_fft::SharedFftBuffer;