sparse.h

Go to the documentation of this file.

/*******************************************************
 * Copyright (c) 2014, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/
 
#pragma once
#include <af/defines.h>
 
#ifdef __cplusplus
namespace af
{
    class array;
 
#if AF_API_VERSION >= 34
    AFAPI array sparse(const dim_t nRows, const dim_t nCols,
                       const array values, const array rowIdx, const array colIdx,
                       const af::storage stype = AF_STORAGE_CSR);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparse(const dim_t nRows, const dim_t nCols, const dim_t nNZ,
                       const void* const values,
                       const int * const rowIdx, const int * const colIdx,
                       const dtype type = f32, const af::storage stype = AF_STORAGE_CSR,
                       const af::source src = afHost);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparse(const array dense, const af::storage stype = AF_STORAGE_CSR);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparseConvertTo(const array in, const af::storage destStrorage);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array dense(const array sparse);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI void sparseGetInfo(array &values, array &rowIdx, array &colIdx, af::storage &stype,
                             const array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparseGetValues(const array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparseGetRowIdx(const array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI array sparseGetColIdx(const array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI dim_t sparseGetNNZ(const array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af::storage sparseGetStorage(const array in);
#endif
}
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_create_sparse_array(
                 af_array *out,
                 const dim_t nRows, const dim_t nCols,
                 const af_array values, const af_array rowIdx, const af_array colIdx,
                 const af_storage stype);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_create_sparse_array_from_ptr(
                 af_array *out,
                 const dim_t nRows, const dim_t nCols, const dim_t nNZ,
                 const void * const values,
                 const int * const rowIdx, const int * const colIdx,
                 const af_dtype type, const af_storage stype,
                 const af_source src);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_create_sparse_array_from_dense(
                 af_array *out, const af_array dense,
                 const af_storage stype);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_convert_to(af_array *out, const af_array in,
                                      const af_storage destStorage);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_to_dense(af_array *out, const af_array sparse);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_info(af_array *values, af_array *rowIdx, af_array *colIdx, af_storage *stype,
                                    const af_array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_values(af_array *out, const af_array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_row_idx(af_array *out, const af_array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_col_idx(af_array *out, const af_array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_nnz(dim_t *out, const af_array in);
#endif
 
#if AF_API_VERSION >= 34
    AFAPI af_err af_sparse_get_storage(af_storage *out, const af_array in);
#endif
 
#ifdef __cplusplus
}
#endif