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

mod layouts;
mod problem_size;

#[macro_export]
macro_rules! testgen_matmul_problem {
    ($kind: ident, $algorithm: ty, $precision: ty, $selection_builder: expr) => {
        mod _problem_generated {
            use super::*;

            pub fn get_selection() -> MatmulSelection {
                $selection_builder.build()
            }
        }

        $crate::testgen_matmul_layouts!(
            $kind,
            $algorithm,
            $precision,
            _problem_generated::get_selection()
        );
    };
}