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

// swap.rs 

use std::mem::swap as memswap;
use crate::types::VecMut; 
use crate::assert_length_eq; 

/// swaps contents of x and y 
///
/// x <-> y 
///
/// args: 
/// * x: [VecMut] - vector x 
/// * y: [VecMut] - vector y 
pub fn swap<T>( 
    mut x: VecMut<'_, T>, 
    mut y: VecMut<'_, T>, 
) { 
    assert_length_eq!(x, y); 

    let x_slice = x.as_slice_mut(); 
    let y_slice = y.as_slice_mut(); 

    for (xv, yv) in x_slice.iter_mut().zip(y_slice.iter_mut()) { 
        memswap(xv, yv); 
    }
}