cubecl-core 0.7.0

CubeCL core create
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use crate::{
    ir::{Arithmetic, ExpandElement, FmaOperator, Instruction, Scope},
    prelude::CubePrimitive,
    unexpanded,
};

/// Fused multiply-add `A*B+C`.
#[allow(unused_variables)]
pub fn fma<C: CubePrimitive>(a: C, b: C, c: C) -> C {
    unexpanded!()
}

/// Expand method of [fma].
#[allow(unused_variables)]
pub fn fma_expand<C: CubePrimitive>(
    scope: &mut Scope,
    a: ExpandElement,
    b: ExpandElement,
    c: ExpandElement,
) -> ExpandElement {
    let output = scope.create_local(a.ty);

    let out = *output;
    let a = *a;
    let b = *b;
    let c = *c;

    scope.register(Instruction::new(
        Arithmetic::Fma(FmaOperator { a, b, c }),
        out,
    ));

    output
}