def Mat.gemm

bend-blas-lapack@0.0.0.1 · mat.bend:137

import Base
import bend-blas-lapack@0.0.0.1/blas.bend as B

137# C = alpha A B + beta C
138def Mat.gemm(+rows: Nat, +k: Nat, +cols: Nat, alpha: F32, a: Mat<rows, k>, b: Mat<k, cols>, beta: F32, c: Mat<rows, cols>) -> IO(Result<&1, &1, U32 & String, Mat<rows, cols>>):
139  MatA{x} = a
140  MatA{y} = b
141  MatA{z} = c
142  IO.bind(Result<&1, &1, U32 & String, Array<F32>>, Result<&1, &1, U32 & String, Mat<rows, cols>>, B.Blas.sgemm(0, 0, U32.from_nat(rows), U32.from_nat(cols), U32.from_nat(k), alpha, x, y, beta, z), r => Mat.wrap(rows, cols, r))