Function matmul

src/backend/opencl/cpu/cpu_sparse_blas.cpp:194–265

////////////////////////////////////////////////////////////////////////////////
template<typename T>
Array<T> matmul(const common::SparseArray<T> lhs, const Array<T> rhs,
                af_mat_prop optLhs, af_mat_prop optRhs) {
    // MKL: CSRMM Does not support optRhs
    UNUSED(optRhs);

    lhs.eval();
    rhs.eval();

    // Similar Operations to GEMM
    sparse_operation_t lOpts = toSparseTranspose(optLhs);

    int lRowDim = (lOpts == SPARSE_OPERATION_NON_TRANSPOSE) ? 0 : 1;
    // int lColDim = (lOpts == SPARSE_OPERATION_NON_TRANSPOSE) ? 1 : 0;

    // Unsupported : (rOpts == SPARSE_OPERATION_NON_TRANSPOSE;) ? 1 : 0;
    static const int rColDim = 1;

    dim4 lDims = lhs.dims();
    dim4 rDims = rhs.dims();
    int M      = lDims[lRowDim];
    int N      = rDims[rColDim];
    // int K = lDims[lColDim];

    Array<T> out = createValueArray<T>(af::dim4(M, N, 1, 1), scalar<T>(0));
    out.eval();

    auto alpha = getScale<T, 1>();
    auto beta  = getScale<T, 0>();

    int ldb = rhs.strides()[1];
    int ldc = out.strides()[1];

    // get host pointers from mapped memory
    mapped_ptr<T> rhsPtr = rhs.getMappedPtr(CL_MAP_READ);
    mapped_ptr<T> outPtr = out.getMappedPtr();

    Array<T> values   = lhs.getValues();
    Array<int> rowIdx = lhs.getRowIdx();
    Array<int> colIdx = lhs.getColIdx();

    mapped_ptr<T> vPtr   = values.getMappedPtr();
    mapped_ptr<int> rPtr = rowIdx.getMappedPtr();
    mapped_ptr<int> cPtr = colIdx.getMappedPtr();
    int *pB              = rPtr.get();
    int *pE              = rPtr.get() + 1;

    sparse_matrix_t csrLhs;
    create_csr_func<T>()(&csrLhs, SPARSE_INDEX_BASE_ZERO, lhs.dims()[0],
                         lhs.dims()[1], pB, pE, cPtr.get(),
                         reinterpret_cast<ptr_type<T>>(vPtr.get()));

    struct matrix_descr descrLhs {};
    descrLhs.type = SPARSE_MATRIX_TYPE_GENERAL;

    mkl_sparse_optimize(csrLhs);

    if (rDims[rColDim] == 1) {
        mkl_sparse_set_mv_hint(csrLhs, lOpts, descrLhs, 1);