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

use cubecl::prelude::Numeric;

/// Implicit convolution spec definiting each element types used in the computation.
pub trait ConvPrecision: Send + Sync + Clone + 'static {
    /// Element type of each input and output tensor of the kernel.
    type EG: Numeric;
    /// Element type of the intermediate representation of the inputs.
    type ES: Numeric;
    /// Element type of the intermediate representation of the output accumulator.
    type EA: Numeric;
}

impl<EG: Numeric, ES: Numeric, EA: Numeric> ConvPrecision for (EG, ES, EA) {
    type EG = EG;
    type ES = ES;
    type EA = EA;
}