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
use std::{fmt::Debug, hash::Hash};

use crate::components::MatrixLayout;

#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq, Default)]
pub struct GlobalMemoryConfig {
    pub elements_in_tile_row: u32,
    pub elements_in_tile_col: u32,
    pub elements_in_stage_row: u32,
    pub elements_in_stage_col: u32,
    pub global_line_size: u32,
    pub check_row_bounds: bool,
    pub check_col_bounds: bool,
    pub matrix_layout: MatrixLayout,
}