pub trait DialectWmmaCompiler<D: Dialect>:
Default
+ Clone
+ Copy
+ Debug
+ Send
+ Sync
+ Eq
+ Hash
+ 'static {
Show 13 methods
// Required methods
fn compile_wmma_fragment_declaration(
f: &mut Formatter<'_>,
var: &Variable<D>,
) -> Result;
fn compile_wmma_instruction(
f: &mut Formatter<'_>,
instruction: &WmmaInstruction<D>,
) -> Result;
fn compile_manual_mma(
f: &mut Formatter<'_>,
mma: ManualMma<'_, D>,
) -> Result;
fn compile_scaled_mma(
f: &mut Formatter<'_>,
mma: ManualMma<'_, D>,
scales_a: Variable<D>,
scales_b: Variable<D>,
scales_factor: u32,
) -> Result;
fn supported_wmma_combinations(
arch: &D::Architecture,
) -> SupportedMmaCombinations;
fn supported_mma_combinations(
arch: &D::Architecture,
) -> SupportedMmaCombinations;
// Provided methods
fn compile_wmma_includes(f: &mut Formatter<'_>, flags: &Flags) -> Result { ... }
fn compile_wmma_type_definitions(
f: &mut Formatter<'_>,
flags: &Flags,
) -> Result { ... }
fn compile_wmma_local_variables(f: &mut Formatter<'_>) -> Result { ... }
fn compile_wwma_fragment_ident(
f: &mut Formatter<'_>,
ident: &FragmentIdent<D>,
) -> Result { ... }
fn compile_wmma_fragment_layout(
f: &mut Formatter<'_>,
layout: &FragmentLayout<D>,
) -> Result { ... }
fn compile_wmma_fragment(
f: &mut Formatter<'_>,
fragment: &Fragment<D>,
) -> Result { ... }
fn supported_scaled_mma_combinations(
_arch: &D::Architecture,
) -> SupportedScaledMmaCombinations { ... }
}Required Methods§
fn compile_wmma_fragment_declaration( f: &mut Formatter<'_>, var: &Variable<D>, ) -> Result
fn compile_wmma_instruction( f: &mut Formatter<'_>, instruction: &WmmaInstruction<D>, ) -> Result
fn compile_manual_mma(f: &mut Formatter<'_>, mma: ManualMma<'_, D>) -> Result
fn compile_scaled_mma( f: &mut Formatter<'_>, mma: ManualMma<'_, D>, scales_a: Variable<D>, scales_b: Variable<D>, scales_factor: u32, ) -> Result
fn supported_wmma_combinations( arch: &D::Architecture, ) -> SupportedMmaCombinations
fn supported_mma_combinations( arch: &D::Architecture, ) -> SupportedMmaCombinations
Provided Methods§
fn compile_wmma_includes(f: &mut Formatter<'_>, flags: &Flags) -> Result
fn compile_wmma_type_definitions(f: &mut Formatter<'_>, flags: &Flags) -> Result
fn compile_wmma_local_variables(f: &mut Formatter<'_>) -> Result
fn compile_wwma_fragment_ident( f: &mut Formatter<'_>, ident: &FragmentIdent<D>, ) -> Result
fn compile_wmma_fragment_layout( f: &mut Formatter<'_>, layout: &FragmentLayout<D>, ) -> Result
fn compile_wmma_fragment( f: &mut Formatter<'_>, fragment: &Fragment<D>, ) -> Result
fn supported_scaled_mma_combinations( _arch: &D::Architecture, ) -> SupportedScaledMmaCombinations
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.