sparse.h (2014-04-30 15:38:09 -0400, (build fd32605))

Go to the documentation of this file.

#pragma once
#include <af/array.h>
#include <af/exception.h>

namespace af {

    typedef enum {
        af_csr,
        af_csc,
        af_coo,
    } af_sparse_t;

#define SPARSE(ty)                                          \
    AFAPI array sparse(const int *row_idx,                  \
                       const int *col_idx,                  \
                       const ty  *values,                   \
                       int nrows, int ncols, int nonzero,   \
                       af_source_t src=afHost,              \
                       unsigned ngfor=0);

    SPARSE(float);
    SPARSE(double);
    SPARSE(cfloat);
    SPARSE(cdouble);
#undef SPARSE

    AFAPI array sparse(array dense);

    AFAPI array sparse(array& row_idx, array& col_idx, array& values, int nrows = -1,
                       int ncols = -1, af_sparse_t format = af_coo);

    AFAPI array dense(array sparse);

    AFAPI afError where(array& I, array& J, array& Val, array sp, af_sparse_t format = af_coo);

    AFAPI array getrow(int nrows, int* rows_idx, array sp);
    AFAPI array getcol(int ncols, int* cols_idx, array sp);
};