Function mtm

src/backend/cpu/sparse_blas.cpp:372–396

template<typename T, bool conjugate>
void mtm(Param<T> output, CParam<T> values, CParam<int> rowIdx,
         CParam<int> colIdx, CParam<T> right, int M, int N, int ldb, int ldc) {
    const T *valPtr   = values.get();
    const int *rowPtr = rowIdx.get();
    const int *colPtr = colIdx.get();
    const T *rightPtr = right.get();
    T *outPtr         = output.get();

    for (int o = 0; o < N; ++o) {
        for (int i = 0; i < M; ++i) { outPtr[i] = scalar<T>(0); }

        for (int i = 0; i < rowIdx.dims(0) - 1; ++i) {
            for (int j = rowPtr[i]; j < rowPtr[i + 1]; ++j) {
                // If stride[0] of right is not 1 then rightPtr[i*stride]
                if (conjugate) {
                    outPtr[colPtr[j]] += getConjugate(valPtr[j]) * rightPtr[i];
                } else {
                    outPtr[colPtr[j]] += valPtr[j] * rightPtr[i];
                }
            }
        }
        rightPtr += ldb;
        outPtr += ldc;
    }
}

template<typename T>
Array<T> matmul(const common::SparseArray<T> &lhs, const Array<T> &rhs,