cubecl_linalg/matmul/components/global/load/loader/
shared.rs

1use cubecl_core as cubecl;
2use cubecl_core::prelude::CubeType;
3
4#[derive(Copy, Clone, CubeType)]
5pub enum BufferId {
6    A,
7    B,
8}
9
10impl BufferId {
11    pub fn to_index(&self) -> u32 {
12        match self {
13            BufferId::A => 0,
14            BufferId::B => 1,
15        }
16    }
17}