half-fma ( 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: half-fma cpu ( a b accumulator -- result )

Methods

USING: cpu.arm.64.features generic kernel math.vectors.simd
math.vectors.simd.extensions
math.vectors.simd.extensions.arm.64.private system ;

M:: arm.64 half-fma ( a b c -- result )
    "fp16" arm64-feature? a half-8? and b half-8? and
    c half-8? and
    [ a b c hfma ]
    [ a b c M\ arm.64 half-fma (call-next-method) ] if ;


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

M: object half-fma [ half-fma-value ] 3map ;