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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

//! Inferred-blueprint smoke tests for TMA routines.

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

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

#[test]
fn simple_tma_cmma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::SimpleTmaCmma(Default::default()),
    );
}

#[test]
fn simple_tma_mma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::SimpleTmaMma(Default::default()),
    );
}

#[test]
fn double_tma_cmma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::DoubleTmaCmma(Default::default()),
    );
}

#[test]
fn double_tma_mma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::DoubleTmaMma(Default::default()),
    );
}

#[test]
fn specialized_tma_cmma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::SpecializedTmaCmma(Default::default()),
    );
}

#[test]
fn specialized_tma_mma() {
    test_matmul_strategy(
        client(),
        square(256, f16_elems()),
        Strategy::SpecializedTmaMma(Default::default()),
    );
}