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

//! Smoke tests for the naive routine.

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

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

#[test]
fn naive_small_f16() {
    test_matmul_strategy(client(), square(16, f16_elems()), Strategy::Naive);
}

#[cfg(feature = "heavy")]
#[test]
fn naive_medium_f32() {
    test_matmul_strategy(client(), square(128, f32_elems()), Strategy::Naive);
}

#[test]
fn naive_odd_shape() {
    test_matmul_strategy(client(), rect(1, 255, 101, f16_elems()), Strategy::Naive);
}