cubecl-tile 0.12.0-pre.1

CubeCL Tile: a space cut into tiles, and the operations on them.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

//! Binding tensors to a kernel and launching it: the [`Launcher`](crate::Launcher), each operand's
//! comptime `spec`, the `binding` that builds it from a tensor, the `arg`s the kernel receives, and
//! the [`Vectorization`] that settles each operand's width.

pub(crate) mod arg;
pub(crate) mod base;
pub(crate) mod binding;
pub(crate) mod refusal;
pub(crate) mod spec;
pub(crate) mod vectorization;

pub use arg::accumulate::{AccumulateArg, AccumulateArgLaunch};
pub use arg::destination::{
    Buffered, Destination, DestinationLaunch, SinkLaunch, SinkLaunchVisitor, TmaStored,
};
pub use arg::last_arrival::LastArrival;
pub use arg::output::{Output, OutputArgs};
pub use arg::relay::Relay;
pub use arg::tma::{TmaBox, TmaLanding, TmaOperand};
pub use base::{Grid, spawnable};
pub use binding::{Arg, Bound, Labelled, Unbound, Unlabelled};
pub use refusal::Refusal;
pub use spec::{BoundaryPolicy, Delivery};
pub use vectorization::{VectorRequirement, Vectorization};