pub trait CubeDispatch:
Clone
+ Copy
+ 'static
+ Send
+ Sync
+ Debug
+ Hash
+ Eq {
// Required methods
fn x_y_indices() -> (u32, u32);
fn batch_index() -> u32;
fn max_x(cube_count: (u32, u32, u32)) -> u32;
fn max_y(cube_count: (u32, u32, u32)) -> u32;
fn max_batches(cube_count: (u32, u32, u32)) -> u32;
fn __expand_x_y_indices(
context: &mut Scope,
) -> <(u32, u32) as CubeType>::ExpandType;
fn __expand_batch_index(
context: &mut Scope,
) -> <u32 as CubeType>::ExpandType;
fn __expand_max_x(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32;
fn __expand_max_y(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32;
fn __expand_max_batches(
context: &mut Scope,
cube_count: (u32, u32, u32),
) -> u32;
}
Expand description
Distributes cube instances across the tensor, assigning each to compute data in distinct regions.
Required Methods§
fn x_y_indices() -> (u32, u32)
fn batch_index() -> u32
fn max_x(cube_count: (u32, u32, u32)) -> u32
fn max_y(cube_count: (u32, u32, u32)) -> u32
fn max_batches(cube_count: (u32, u32, u32)) -> u32
fn __expand_x_y_indices( context: &mut Scope, ) -> <(u32, u32) as CubeType>::ExpandType
fn __expand_batch_index(context: &mut Scope) -> <u32 as CubeType>::ExpandType
fn __expand_max_x(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32
fn __expand_max_y(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32
fn __expand_max_batches(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32
Dyn Compatibility§
This trait is not dyn compatible.
In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.