Function mtv

src/backend/opencl/cpu/cpu_sparse_blas.cpp:324–351

template<typename T, bool conjugate>
void mtv(Array<T> output, const Array<T> values, const Array<int> rowIdx,
         const Array<int> colIdx, const Array<T> right, int M) {
    mapped_ptr<T> oPtr   = output.getMappedPtr();
    mapped_ptr<T> rhtPtr = right.getMappedPtr();
    mapped_ptr<T> vPtr   = values.getMappedPtr();
    mapped_ptr<int> rPtr = rowIdx.getMappedPtr();
    mapped_ptr<int> cPtr = colIdx.getMappedPtr();

    T const *const valPtr   = vPtr.get();
    int const *const rowPtr = rPtr.get();
    int const *const colPtr = cPtr.get();
    T const *const rhsPtr   = rhtPtr.get();
    T *const outPtr         = oPtr.get();

    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 rhsPtr[i*stride]
            if (conjugate) {
                outPtr[colPtr[j]] =
                    outPtr[colPtr[j]] + getConjugate(valPtr[j]) * rhsPtr[i];
            } else {
                outPtr[colPtr[j]] = outPtr[colPtr[j]] + valPtr[j] * rhsPtr[i];
            }
        }
    }
}

template<typename T, bool conjugate>
void mm(Array<T> output, const Array<T> values, const Array<int> rowIdx,