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

Go to the documentation of this file.

#ifndef LAMA_OPENMP_JDS_UTILS_HPP_
#define LAMA_OPENMP_JDS_UTILS_HPP_

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

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT OpenMPJDSUtils
{
public:

    template<typename ValueType>
    static void setDiagonalWithScalar( const IndexType numDiagonal,
                                       ValueType values[],
                                       Scalar scalar );

    template<typename ValueType, typename OtherValueType>
    static void scaleValue( const IndexType numRows,
                            const IndexType perm[],
                            const IndexType ilg[],
                            const IndexType dlg[],
                            ValueType mValues[],
                            const OtherValueType values[] );

    template<typename ValueType, typename OtherValueType>
    static void getRow( OtherValueType row[],
                        const IndexType i,
                        const IndexType numColumns,
                        const IndexType numRows,
                        const IndexType perm[],
                        const IndexType ilg[],
                        const IndexType dlg[],
                        const IndexType ja[],
                        const ValueType values[] );

    template<typename ValueType, typename NoType>
    static ValueType getValue( const IndexType i,
                               const IndexType j,
                               const IndexType numRows,
                               const IndexType* dlg,
                               const IndexType* ilg,
                               const IndexType* perm,
                               const IndexType* ja,
                               const ValueType* values );

    static bool checkDiagonalProperty( const IndexType numDiagonals,
                                       const IndexType numRows,
                                       const IndexType numColumns,
                                       const IndexType perm[],
                                       const IndexType ja[],
                                       const IndexType dlg[] );

    static bool check( const IndexType numRows,
                       const IndexType numValues,
                       const IndexType numColumns,
                       const IndexType ja[],
                       const IndexType ilg[],
                       const IndexType dlg[] );

    static void sortRows( IndexType array[], IndexType perm[], const IndexType n );

    static void setInversePerm( IndexType inversePerm[], 
                                const IndexType perm[], 
                                const IndexType n );

    static IndexType ilg2dlg( IndexType dlg[],
                              const IndexType numDiagonals,
                              const IndexType ilg[],
                              const IndexType numRows );

    template<typename JDSValueType, typename CSRValueType>
    static void getCSRValues( IndexType csrJA[],
                              CSRValueType csrValues[],
                              const IndexType csrIA[],
                              const IndexType numRows,
                              const IndexType jdsPerm[],
                              const IndexType jdsILG[],
                              const IndexType jdsDLG[],
                              const IndexType jdsJA[],
                              const JDSValueType jdsValues[] );

    template<typename JDSValueType, typename CSRValueType>
    static void setCSRValues( IndexType jdsJA[],
                              JDSValueType jdsValues[],
                              const IndexType numRows,
                              const IndexType jdsPerm[],
                              const IndexType jdsILG[],
                              const IndexType jdsDLG[],
                              const IndexType csrIA[],
                              const IndexType csrJA[],
                              const CSRValueType csrValues[] );

    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 perm[],
                            const IndexType jdsILG[],
                            const IndexType ndlg,
                            const IndexType jdsDLG[],
                            const IndexType jdsJA[],
                            const ValueType jdsValues[],
                            SyncToken* syncToken );

    template <typename ValueType>
    static void jacobi( ValueType solution[],
                        const IndexType numRows,
                        const IndexType jdsPerm[],
                        const IndexType jdsIlg[],
                        const IndexType jdsNumDiagonals,
                        const IndexType jdsDlg[],
                        const IndexType jdsJA[],
                        const ValueType jdsValues[],
                        const ValueType oldSolution[],
                        const ValueType rhs[],
                        const ValueType omega,
                        SyncToken* syncToken );

    template<typename ValueType>
    static void jacobiHalo( ValueType solution[],
                            const IndexType numRows,
                            const ValueType localDiagonal[],
                            const IndexType numDiagonals,
                            const IndexType jdsHaloPerm[],
                            const IndexType jdsHaloIlg[],
                            const IndexType jdsHaloDlg[],
                            const IndexType jdsHaloJA[],
                            const ValueType jdsHaloValues[],
                            const ValueType oldSolution[],
                            const ValueType omega, 
                            SyncToken* syncToken );

    static void setInterface( struct JDSUtilsInterface& JDSUtils );

private:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

};

/* --------------------------------------------------------------------------- */

}  // namespace lama

#endif //  LAMA_OPENMP_JDS_UTILS_HPP_