Skip to main content

fma

strided_basic

Function fma 

Source 
pub fn fma<D, A, B, OpA, OpB>(
    dest: &mut StridedViewMut<'_, D>,
    a: &StridedView<'_, A, OpA>,
    b: &StridedView<'_, B, OpB>,
    exec: Exec<'_>,
) -> Result<()>
where
    A: Copy + Mul<B, Output = D> + Send + Sync,
    B: Copy + Send + Sync,
    D: Copy + Add<D, Output = D> + Send + Sync,
    OpA: ElementOp<A>,
    OpB: ElementOp<B>,
Expand description

Fused multiply-add: dest[i] += OpA::apply(a[i]) * OpB::apply(b[i]).

Operands may have different element types. Element operations are applied lazily.