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

Go to the documentation of this file.

#ifndef LAMA_OPENMP_DIA_UTILS_HPP_
#define LAMA_OPENMP_DIA_UTILS_HPP_

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT OpenMPDIAUtils
{
public:

    template<typename ValueType>
    static void getCSRSizes( IndexType csrSizes[],
                             bool diagonalFlag,
                             const IndexType numRows,
                             const IndexType numColumns,
                             const IndexType numDiagonals,
                             const IndexType diaOffsets[],
                             const ValueType diaValues[],
                             const ValueType eps );

    template<typename DIAValueType, typename CSRValueType>
    static void getCSRValues( IndexType csrJA[],
                              CSRValueType csrValues[],
                              const IndexType csrIA[],
                              const bool diagonalFlag,
                              const IndexType numRows,
                              const IndexType numColumns,
                              const IndexType numDiagonals,
                              const IndexType diaOffsets[],
                              const DIAValueType diaValues[],
                              const DIAValueType eps );

    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 numColumns,
                            const IndexType numDiagonals,
                            const IndexType diaOffsets[],
                            const ValueType diaValues[],
                            SyncToken* syncToken );

    template <typename ValueType>
    static void jacobi( ValueType solution[],
                        const IndexType numColumns,
                        const IndexType numDiagonals,
                        const IndexType diaOffset[],
                        const ValueType diaValues[],
                        const ValueType oldSolution[],
                        const ValueType rhs[],
                        const ValueType omega,
                        const IndexType numRows,
                        SyncToken* syncToken );

    static void setInterface( struct DIAUtilsInterface& DIAUtils );

private:

    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 numColumns,
                            const IndexType numDiagonals,
                            const IndexType diaOffsets[],
                            const ValueType diaValues[]  );

    LAMA_LOG_DECL_STATIC_LOGGER(logger);
};

}  // namespace lama

#endif //  LAMA_OPENMP_DIA_UTILS_HPP_