lak-kernels 0.1.1

BLAS-like linear algebra kernels in fully-safe Rust.
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
31

// common.rs

use std::ops::Sub;
use lak::traits::{Abs, Max, TestFloat}; 

/// how many times each routine is tested 
/// using a seeded random number generator
pub const ITERATIONS: u64 = 4; 

pub fn assert_close<T>( 
    a: &[T], 
    b: &[T], 
) 
where 
    T: TestFloat 
        + Sub<Output = T> 
        + Abs
        + Max, 
{ 
    assert_eq!(a.len(), b.len());

    for (i, (&x, &y)) in a.iter().zip(b.iter()).enumerate() { 
        let delta = (x - y).abs(); 
        let tolerance = T::ATOL + T::RTOL * x.abs().max(y.abs()); 

        assert!( 
            delta <= tolerance, 
            "mismatch at idx {i}: {x} vs. {y} / δ={delta}, ϵ={tolerance}"
        ); 
    }
}