/home/brandes/workspace/LAMA/src/lama/openmp/OpenMPCOOUtils.hpp

Go to the documentation of this file.

#ifndef LAMA_OPENMP_COO_UTILS_HPP_
#define LAMA_OPENMP_COO_UTILS_HPP_

// for dll_import
#include <lama/config.hpp>

// others
#include <lama/LAMATypes.hpp>
#include <lama/SyncToken.hpp>

// assert
#include <lama/exception/LAMAAssert.hpp>

// logging
#include <logging/logging.hpp>

namespace lama
{

class LAMA_DLL_IMPORTEXPORT OpenMPCOOUtils
{
public:

    static void getCSRSizes( IndexType csrSizes[],
                             const IndexType numRows,
                             const IndexType numValues,
                             const IndexType cooIA[] );

    template<typename COOValueType, typename  CSRValueType>
    static void getCSRValues( IndexType csrJA[],
                              CSRValueType csrValues[],
                              IndexType csrIA[],
                              const IndexType numRow,
                              const IndexType numValues,
                              const IndexType cooIA[],
                              const IndexType cooJA[],
                              const COOValueType cooValues[] );

    template<typename COOValueType, typename CSRValueType>
    static void setCSRValues( IndexType cooIA[],
                              IndexType cooJA[],
                              COOValueType cooValues[],
                              const IndexType numRows,
                              const IndexType numDiagonals,
                              const IndexType csrIA[],
                              const IndexType csrJA[],
                              const CSRValueType csrValues[],
                              const bool csrDiagonalProperty );

    template <typename ValueType>
    static void normalGEMV( ValueType result[],
                            const ValueType alpha,
                            const ValueType x[],
                            const ValueType beta,
                            const ValueType y[],
                            const IndexType numRows,
                            const IndexType cooIA[],
                            const IndexType cooJA[],
                            const ValueType cooValues[],
                            const IndexType numValues,
                            SyncToken* syncToken );

    template <typename ValueType>
    static void jacobi( ValueType* solution,
                        const IndexType cooNumValues,
                        const IndexType cooIA[],
                        const IndexType cooJA[],
                        const ValueType cooValues[],
                        const ValueType oldSolution[],
                        const ValueType rhs[],
                        const ValueType omega,
                        const IndexType numRows,
                        SyncToken* syncToken );

    static void setInterface( struct COOUtilsInterface& COOUtils );

private:

    LAMA_LOG_DECL_STATIC_LOGGER( logger );
};

}  // namespace lama

#endif //  LAMA_COO_STORAGE_UTILS_HPP_