tenferro-gpu 0.3.0

CubeCL-backed CUDA and WebGPU provider backends for tenferro tensors.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

//! Internal CubeCL kernels and launch helpers for tenferro.
//!
//! This crate owns GPU kernel definitions but does not own tenferro tensor
//! values, device placement, or backend dispatch.

#[cfg(feature = "cuda")]
pub(crate) mod error;
#[cfg(feature = "cuda")]
pub(crate) mod reduce;

#[cfg(feature = "cuda")]
pub(crate) mod diagonal;
#[cfg(feature = "cuda")]
pub(crate) mod elementwise;
mod helpers;
#[cfg(feature = "cuda")]
pub(crate) mod indexing;
pub(crate) mod structural;

#[cfg(feature = "cuda")]
pub(crate) use error::{CubeclKernelError, Result};