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

//! What a tile holds ([`TileKind`]): the one enum every verb on a tile dispatches on.

use cubecl::prelude::*;

use crate::*;

/// A tile's backing store.
/// `Clone` copies the handle, not the cells: sound only where nothing rewrites the buffer.
#[allow(clippy::large_enum_variant)]
#[expect(
    dead_code,
    reason = "built through the expand type's generated constructors"
)]
#[derive(CubeType, Clone)]
#[expand(derive(Clone))]
pub(crate) enum TileKind<T: Numeric> {
    /// An addressable buffer with a window, in global or shared memory.
    Memory(Memory<T>),
    /// One plane-level tile, sliced across the plane's units and never addressable.
    PlaneTile(PlaneTile<T>),
    /// The grid of plane tiles one plane owns; only comptime coordinates select through it.
    PlanePartition(PlanePartition<T>),
    /// A TMA tensor-map source, copied only in bulk into shared memory.
    TmaGmem(TmaData<T>),
    /// A read-only source evaluated from logical coordinates with no backing buffer.
    Procedural(Procedural<T>),
    /// A tile the plane holds in its units, one line to a unit ([`Lines`]).
    Lines(Lines<T>),
}