(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: cpu.arm.64.features generic kernel
math.vectors.simd.extensions
math.vectors.simd.extensions.arm.64.private system ;

M: arm.64 (vbdot2+)
    \ bdot2 arm64-kernel-supported?
    [ bdot2 ] [ M\ arm.64 (vbdot2+) (call-next-method) ] if ;


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 ;