Skip to main content

cubecl_tile/launch/
mod.rs

1//! Binding tensors to a kernel and launching it: the [`Launcher`](crate::Launcher), each operand's
2//! comptime `spec`, the `binding` that builds it from a tensor, the `arg`s the kernel receives, and
3//! the [`Vectorization`] that settles each operand's width.
4
5pub(crate) mod arg;
6pub(crate) mod base;
7pub(crate) mod binding;
8pub(crate) mod refusal;
9pub(crate) mod spec;
10pub(crate) mod vectorization;
11
12pub use arg::accumulate::{AccumulateArg, AccumulateArgLaunch};
13pub use arg::destination::{
14    Buffered, Destination, DestinationLaunch, SinkLaunch, SinkLaunchVisitor, TmaStored,
15};
16pub use arg::last_arrival::LastArrival;
17pub use arg::output::{Output, OutputArgs};
18pub use arg::relay::Relay;
19pub use arg::tma::{TmaBox, TmaLanding, TmaOperand};
20pub use base::{Grid, spawnable};
21pub use binding::{Arg, Bound, Labelled, Unbound, Unlabelled};
22pub use refusal::Refusal;
23pub use spec::{BoundaryPolicy, Delivery};
24pub use vectorization::{VectorRequirement, Vectorization};