2d-matmul-simd ( vec1: float-array vec2: float-array res: float-array m: fixnum n: fixnum p: fixnum -- )

Vocabulary

tensors.private

Inputs

vec1	a float-array
vec2	a float-array
res	a float-array
m	a fixnum
n	a fixnum
p	a fixnum

Outputs

None

Definition

USING: accessors alien.data kernel math math.vectors
math.vectors.simd sequences sequences.private typed ;

IN: tensors.private

TYPED:: 2d-matmul-simd
    ( vec1: float-array vec2: float-array res: float-array m: fixnum n: fixnum p: fixnum -- )
    { n p } vec2 <tensor> 2d-transpose vec>> :> vec2 m [
        :> i i n * :> in i p * :> ip vec1 in n make-subseq
        float-4 cast-array p [
            :> j dup vec2 j n * n make-subseq float-4 cast-array
            0.0 [ vdot + ] 2reduce ip j + res set-nth-unsafe
        ] each-integer drop
    ] each-integer ;