lak-kernels 0.1.0

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

// scal.rs

use std::ops::{Mul, MulAssign};

use crate::types::VecMut; 

/// scales x by alpha 
///
/// x <- alpha * x
///
/// args: 
/// * alpha: T - scales x 
/// * x: [VecMut] - vector x 
pub fn scal<T>( 
    alpha: T, 
    mut x: VecMut<'_, T>
) 
where 
    T: Mul<Output=T>
        + Copy 
        + MulAssign, 
{ 
    let x_slice = x.as_slice_mut(); 

    // no simd needed, already fast 
    for x_value in x_slice.iter_mut() { 
        *x_value *= alpha 
    }
}