Trait CubeDispatch

Source
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§

Source

fn x_y_indices() -> (u32, u32)

Source

fn batch_index() -> u32

Source

fn max_x(cube_count: (u32, u32, u32)) -> u32

Source

fn max_y(cube_count: (u32, u32, u32)) -> u32

Source

fn max_batches(cube_count: (u32, u32, u32)) -> u32

Source

fn __expand_x_y_indices( context: &mut Scope, ) -> <(u32, u32) as CubeType>::ExpandType

Source

fn __expand_batch_index(context: &mut Scope) -> <u32 as CubeType>::ExpandType

Source

fn __expand_max_x(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32

Source

fn __expand_max_y(context: &mut Scope, cube_count: (u32, u32, u32)) -> u32

Source

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.

Implementors§