rublas 0.1.21

Linear algebra and matrix multiplication for Ruda.
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

#![allow(missing_docs)]

pub mod basic;
#[cfg(feature = "extended")]
pub mod extended;
#[cfg(feature = "full")]
pub mod full;

mod bias;
pub(crate) mod launcher_strategy;

pub(crate) use launcher_strategy::test_matmul_strategy;

#[cfg(feature = "extended")]
pub(crate) use extended::test_matmul_test_strategy;

pub(crate) use rublas::kernel_ir::cpu_reference::assert_result;

use ruda_kernel::tiling::MatrixLayout;
use ruda_test_utils::StrideSpec;

pub(crate) fn layout_to_stride_spec(layout: MatrixLayout) -> StrideSpec {
    match layout {
        MatrixLayout::RowMajor => StrideSpec::RowMajor,
        MatrixLayout::ColMajor => StrideSpec::ColMajor,
    }
}