Skip to main content

rublas/
lib.rs

1//! Ruda BLAS operations and device tensor kernels.
2
3#[cfg(feature = "cann")]
4pub mod cann;
5
6
7
8#[cfg(feature = "kernel-ir")]
9#[allow(unsafe_code)]
10pub mod kernel_ir;
11
12#[cfg(feature = "tensor-vector")]
13#[allow(unsafe_code)]
14pub mod tensor_vector;
15
16#[cfg(feature = "tensor-matmul")]
17pub mod tensor_matmul;
18
19#[cfg(feature = "tensor-int4")]
20pub mod tensor_int4;
21
22#[cfg(feature = "tensor-nf4")]
23#[allow(unsafe_code)]
24pub mod tensor_nf4;
25
26#[cfg(feature = "tensor-grouped")]
27pub mod tensor_grouped;