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
28
29
30

//! Inferred-blueprint smoke tests for unit-based routines.
//!
//! The interleaved routine is not covered here because its tile matmul
//! requires `tile.k % plane_dim == 0`, which the inferred selector doesn't
//! enforce — a forced-blueprint variant lives in `extended/tiling_scheme.rs`.

use rublas::kernel_ir::launch::Strategy;

use super::common::{client, f16_elems, square};
use crate::suite::test_matmul_strategy;

#[cfg(feature = "heavy")]
#[test]
fn simple_unit() {
    test_matmul_strategy(
        client(),
        square(64, f16_elems()),
        Strategy::SimpleUnit(Default::default()),
    );
}

#[cfg(feature = "heavy")]
#[test]
fn double_unit() {
    test_matmul_strategy(
        client(),
        square(64, f16_elems()),
        Strategy::DoubleUnit(Default::default()),
    );
}