burn-jit 0.16.1

Generic backend that can be compiled just-in-time to any shader language target
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

use super::args::FusedMatmulArgs;
use cubecl::{linalg::matmul::components::MatmulSpec, prelude::Numeric};
use std::marker::PhantomData;

/// Specification for a fused standard matmul.
#[derive(Clone)]
pub struct FusedMatmulSpec<EG: Numeric, ES: Numeric, EA: Numeric> {
    _eg: PhantomData<EG>,
    _es: PhantomData<ES>,
    _ea: PhantomData<EA>,
}

impl<EG: Numeric, ES: Numeric, EA: Numeric> MatmulSpec for FusedMatmulSpec<EG, ES, EA> {
    type EG = EG;
    type ES = ES;
    type EA = EA;
    type Args = FusedMatmulArgs;
}