(vbdot2+) ( a b accumulator -- result )

Vocabulary

math.vectors.simd.extensions

Inputs

a	an object
b	an object
accumulator	an object

Outputs

result

an object

Definition

USING: system ;

IN: math.vectors.simd.extensions

HOOK: (vbdot2+) cpu ( a b accumulator -- result )

Methods

USING: kernel math math.floats.small.bfloat math.vectors.simd
math.vectors.simd.extensions sequences ;

M:: object (vbdot2+) ( a b accumulator -- result )
    4 <iota> [| lane |
        lane accumulator nth lane 2 * a nth lane 2 * 1 + a nth
        lane 2 * b nth lane 2 * 1 + b nth bfloat-dot-add
    ] float-4 new map-as ;