cubecl_cpp::shared

Trait DialectWmmaCompiler

Source 
pub trait DialectWmmaCompiler<D: Dialect>:
    Default
    + Clone
    + Copy
    + Debug
    + Send
    + Sync
    + Eq
    + Hash
    + 'static {
    type Architecture: Architecture;

    // Required methods
    fn compile_wmma_includes(f: &mut Formatter<'_>) -> Result;
    fn compile_wmma_type_definitions(f: &mut Formatter<'_>) -> Result;
    fn compile_local_variables(f: &mut Formatter<'_>) -> Result;
    fn compile_fragment_ident(
        ident: &FragmentIdent<D>,
        f: &mut Formatter<'_>,
    ) -> Result;
    fn compile_fragment_layout(
        layout: &FragmentLayout<D>,
        f: &mut Formatter<'_>,
    ) -> Result;
    fn compile_fragment(fragment: &Fragment<D>, f: &mut Formatter<'_>) -> Result;
    fn compile_instruction(
        instruction: &WmmaInstruction<D>,
        f: &mut Formatter<'_>,
    ) -> Result;
    fn supported_wmma_combinations(
        arch: &Self::Architecture,
    ) -> SupportedWmmaCombinations;
}

Required Associated Types§

Source

type Architecture: Architecture

Required Methods§

Source

fn compile_wmma_includes(f: &mut Formatter<'_>) -> Result

Source

fn compile_wmma_type_definitions(f: &mut Formatter<'_>) -> Result

Source

fn compile_local_variables(f: &mut Formatter<'_>) -> Result

Source

fn compile_fragment_ident( ident: &FragmentIdent<D>, f: &mut Formatter<'_>, ) -> Result

Source

fn compile_fragment_layout( layout: &FragmentLayout<D>, f: &mut Formatter<'_>, ) -> Result

Source

fn compile_fragment(fragment: &Fragment<D>, f: &mut Formatter<'_>) -> Result

Source

fn compile_instruction( instruction: &WmmaInstruction<D>, f: &mut Formatter<'_>, ) -> Result

Source

fn supported_wmma_combinations( arch: &Self::Architecture, ) -> SupportedWmmaCombinations

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§