cubek-tile 0.3.0

CubeK: Tile
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
29

//! A unit's place in the team it shares a leaf with.

use cubecl::prelude::*;

/// A unit's index in its team and the team's unit count.
#[derive(CubeType, Clone)]
#[expand(derive(Clone))]
pub struct TeamUnit {
    /// This unit, among the team's.
    pub index: usize,
    /// Units the team holds.
    pub units: usize,
}

#[cube]
impl TeamUnit {
    /// This unit in a team of `units`, at `index`.
    pub fn new(index: usize, units: usize) -> TeamUnit {
        TeamUnit { index, units }
    }

    /// A team laid along the cube's x dim: one team per row of the cube.
    pub fn along_x() -> TeamUnit {
        TeamUnit {
            index: UNIT_POS_X as usize,
            units: CUBE_DIM_X as usize,
        }
    }
}