(validate-gemm) ( A B C -- )

Vocabulary

math.blas.matrices.private

Inputs

A	an object
B	an object
C	an object

Outputs

None

Definition

USING: combinators.short-circuit kernel math.blas.matrices ;

IN: math.blas.matrices.private

: (validate-gemm) ( A B C -- )
    {
        [ [ Mwidth ] [ Mheight ] [ drop ] tri* = ]
        [ [ Mheight ] [ drop ] [ Mheight ] tri* = ]
        [ [ drop ] [ Mwidth ] [ Mwidth ] tri* = ]
    } 3&&
    [ "Mismatched matrices in matrix multiplication" throw ]
    unless ;