replace remaining gemm calls