ruda-kernel 0.1.2

Ruda Rust kernel DSL and device tensor operations.
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
25
26
27
28
29
30

#[cfg(feature = "quantization-kernels")]
pub mod dequantize;

#[cfg(feature = "quantization-kernels")]
pub mod quantize;

#[cfg(feature = "quantization-kernels")]
pub mod layout;

pub use ruda_core::quant::scheme;

#[cfg(feature = "quantization-kernels")]
pub(crate) mod utils {
    use crate::quantization::scheme::{QuantLevel, QuantScheme};

    pub(crate) fn check_block_size_compat(scheme: &QuantScheme, div: usize) {
        // Validate block size compatibility
        if let QuantScheme {
            level: QuantLevel::Block(block_size),
            ..
        } = scheme
        {
            let block_size = *block_size.as_slice().last().unwrap() as usize;
            assert!(
                block_size.is_multiple_of(div),
                "Block size must be divisible by {div}, got block_size={block_size}"
            );
        }
    }
}