cubecl_linalg::matmul::components::global

Trait OutputLoader

Source 
pub trait OutputLoader<EO: Numeric>:
    CubeType
    + 'static
    + Send
    + Sync {
    type StageWriter: StageWriter<EO>;

    // Required methods
    fn as_stage_writer<G: GlobalConfig>(unloader: Self) -> Self::StageWriter;
    fn __expand_as_stage_writer<G: GlobalConfig>(
        context: &mut Scope,
        unloader: <Self as CubeType>::ExpandType,
    ) -> <Self::StageWriter as CubeType>::ExpandType;
}
Expand description

Output to the global matmul

§Note

It is only a wrapper over the stage writer because there is no K for the output. Could be deleted in favor of having only the StageWriter

Required Associated Types§

Source

type StageWriter: StageWriter<EO>

Required Methods§

Source

fn as_stage_writer<G: GlobalConfig>(unloader: Self) -> Self::StageWriter

Source

fn __expand_as_stage_writer<G: GlobalConfig>( context: &mut Scope, unloader: <Self as CubeType>::ExpandType, ) -> <Self::StageWriter as CubeType>::ExpandType

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§