mircuda 0.3.0

Native, explicit-stream Rust gateway to NVIDIA CUDA
1
2
3
4
5
6
7
8
9
10
11
mod embedding;
mod gathered;
mod matmul;
#[cfg(feature = "cutlass")]
mod tensor_core;

pub use embedding::{MxFp8Embedding, MxFp8EmbeddingOperands, MxFp8EmbeddingSpec};
pub use gathered::{MxFp8Gathered, MxFp8GatheredOperands, MxFp8GatheredSpec};
pub use matmul::{MxFp8Matmul, MxFp8Spec};
#[cfg(feature = "cutlass")]
pub use tensor_core::{MxFp8TensorCore, MxFp8TensorCoreScratch};