blas-lapack-rs / cblas Goto Github PK

extern crate cblas;
extern crate intel_mkl_src;

use cblas::*;

fn main() {
    let (m, n, k) = (1024, 1024, 1024);
    let a: Vec<f32> = (0..(m * k)).map(|i| i as f32).collect();
    let b: Vec<f32> = (0..(k * n)).map(|i| i as f32).collect();
    let mut c: Vec<f32> = (0..(m * n)).map(|_| 0.0).collect();
    unsafe {
        sgemm(
            Layout::RowMajor,
            Transpose::None,
            Transpose::None,
            m,
            n,
            k,
            1.0,
            &a,
            k,
            &b,
            n,
            0.0,
            &mut c,
            n,
        )
    }
}

// extern crate openblas_src; // Same results as with netlib
extern crate netlib_src;
fn main() {
    let a_slice = vec![
        1., 1.,
        1., 1.,
        1., 1.,
        1., 1.
    ];
    let mut c_slice = vec![
        0., 0.,
        0., 0.
    ];

    unsafe {
        cblas::ssyrk(
            cblas::Layout::RowMajor, // change to cblas::Layout::ColumnMajor for correct result
            cblas::Part::Upper,
            cblas::Transpose::Ordinary,
            2 as i32,
            4 as i32,
            1.0,
            &a_slice,
            4 as i32,
            0.0,
            &mut c_slice,
            2 as i32,
        );
    }

    println!("{:?}", c_slice);
}

extern crate cblas;

use cblas::*;

fn main() {

    let (m, n, k) = (2, 4, 3);
    let a = vec![
        1.0, 4.0,
        2.0, 5.0,
        3.0, 6.0,
    ];
    let b = vec![
        1.0, 5.0,  9.0,
        2.0, 6.0, 10.0,
        3.0, 7.0, 11.0,
        4.0, 8.0, 12.0,
    ];
    let mut c = vec![
        2.0, 7.0,
        6.0, 2.0,
        0.0, 7.0,
        4.0, 2.0,
    ];

    unsafe {
        dgemm(Layout::ColumnMajor, Transpose::None, Transpose::None,
              m, n, k, 1.0, &a, m, &b, k, 1.0, &mut c, m);
    }

    assert!(
        c == vec![
            40.0,  90.0,
            50.0, 100.0,
            50.0, 120.0,
            60.0, 130.0,
        ]
    )
}