cubecl-matmul 0.8.1

CubeCL Matrix Multiplication Kernels Engine
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

mod hypercube;
mod partition_buffering;
mod specialized;

#[macro_export]
macro_rules! testgen_matmul_advanced {
    ($kind: ident, $algorithm: ty, $precision: ty, $tiling_scheme_builder: expr) => {
        use $crate::components::{MatmulSelection, MatmulSelectionBuilder};

        mod _advanced {
            use super::*;

            pub fn get_selection_builder() -> MatmulSelectionBuilder {
                let tiling_scheme = $tiling_scheme_builder.build().unwrap();
                let client = TestRuntime::client(&Default::default());
                let plane_dim = client.properties().hardware.plane_size_max;
                MatmulSelection::builder(tiling_scheme, plane_dim)
            }
        }

        $crate::testgen_matmul_specialized!(
            $kind,
            $algorithm,
            $precision,
            _advanced::get_selection_builder()
        );
    };
}