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

Go to the documentation of this file.

#ifndef LAMA_OPENMP_ELL_UTILS_HPP_
#define LAMA_OPENMP_ELL_UTILS_HPP_

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

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

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT OpenMPELLUtils
{
public:

    static IndexType countNonEmptyRowsBySizes( const IndexType sizes[], const IndexType numRows );

    static void setNonEmptyRowsBySizes( IndexType rowIndexes[], const IndexType numNonEmptyRows,
                                        const IndexType sizes[], const IndexType numRows );

    static inline IndexType ellindex( const IndexType i, const IndexType jj, const IndexType numRows )
    {
        return jj * numRows + i;
    }

    static void check ( const IndexType mNumRows,
                        const IndexType mNumValuesPerRow,
                        const IndexType mNumColumns,
                        const IndexType *ia,
                        const IndexType *ja,
                        const char* msg );



    template<typename ValueType, typename OtherValueType>
    static void getRow( OtherValueType *row,
                        const IndexType i,
                        const IndexType numRows,
                        const IndexType numColumns,
                        const IndexType *ia,
                        const IndexType *ja,
                        const ValueType *values );

    template<typename ValueType, typename OtherValueType>
    static OtherValueType getValue ( const IndexType i,
                                     const IndexType j,
                                     const IndexType numRows,
                                     const IndexType *ia,
                                     const IndexType *ja,
                                     const ValueType *values );

    static bool hasDiagonalProperty( const IndexType numDiagonals, const IndexType csrJA[] );

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

    template<typename ValueType>
    static void compressIA( const IndexType IA[],
                            const IndexType JA[],
                            const ValueType values[],
                            const IndexType numRows,
                            const ValueType eps,
                            IndexType newIA[] );

    template<typename ValueType>
    static void compressValues( const IndexType IA[],
                                const IndexType JA[],
                                const ValueType values[],
                                const IndexType numRows,
                                const ValueType eps,
                                IndexType newJA[],
                                ValueType newValues[] );

    template<typename ELLValueType, typename CSRValueType>
    static void getCSRValues( IndexType csrJA[],
                              CSRValueType csrValues[],
                              const IndexType csrIA[],
                              const IndexType numRows,
                              const IndexType ellSizes[],
                              const IndexType ellJA[],
                              const ELLValueType ellValues[] );

    template<typename ELLValueType, typename CSRValueType>
    static void setCSRValues( IndexType ellJA[],
                              ELLValueType ellValues[],
                              const IndexType ellSizes[],
                              const IndexType numRows,
                              const IndexType numValuesPerRow,
                              const IndexType csrIA[],
                              const IndexType csrJA[],
                              const CSRValueType csrValues[] );

    template <typename ValueType>
    static void computeIA( const IndexType aIA[],
                           const IndexType aJA[],
                           const IndexType aNumRows,
                           const IndexType bIA[],
                           const IndexType bJA[],
                           const IndexType bNumRows,
                           IndexType cIA[] );

    template <typename ValueType>
    static void computeValues( const IndexType aIA[],
                               const IndexType aJA[],
                               const ValueType aValues[],
                               const IndexType aNumRows,
                               const IndexType bIA[],
                               const IndexType bJA[],
                               const ValueType bValues[],
                               const IndexType bNumRows,
                               const ValueType alpha,
                               const IndexType cIA[],
                               IndexType cJA[],
                               ValueType cValues[] );

    template <typename ValueType>
    static void addComputeIA( const IndexType aIA[],
                              const IndexType aJA[],
                              const IndexType aNumRows,
                              const IndexType bIA[],
                              const IndexType bJA[],
                              const IndexType bNumRows,
                              IndexType cIA[] );

    template <typename ValueType>
    static void addComputeValues( const IndexType aIA[],
                                  const IndexType aJA[],
                                  const ValueType aValues[],
                                  const IndexType aNumRows,
                                  const IndexType bIA[],
                                  const IndexType bJA[],
                                  const ValueType bValues[],
                                  const IndexType bNumRows,
                                  const ValueType beta,
                                  const IndexType cIA[],
                                  IndexType cJA[],
                                  ValueType cValues[] );

    template <typename ValueType>
    static void jacobi( ValueType solution[],
                        const IndexType numRows,
                        const IndexType ellNumValuesPerRow,
                        const IndexType ellSizes[],
                        const IndexType ellJA[],
                        const ValueType ellValues[],
                        const ValueType oldSolution[],
                        const ValueType rhs[],
                        const ValueType omega,
                        class SyncToken* syncToken );

    template<typename ValueType>
    static void jacobiHalo( ValueType solution[],
                            const IndexType numRows,
                            const ValueType diagonal[],
                            const IndexType ellNumValuesPerRow,
                            const IndexType ellSizes[],
                            const IndexType ellJA[],
                            const ValueType ellValues[],
                            const IndexType rowIndexes[],
                            const IndexType numNonEmptyRows,
                            const ValueType oldSolution[],
                            const ValueType omega,
                            class SyncToken* syncToken );

    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 numNonZerosPerRow,
                            const IndexType csrIA[],
                            const IndexType csrJA[],
                            const ValueType csrValues[],
                            class SyncToken* syncToken );

    template <typename ValueType>
    static void sparseGEMV( ValueType result[],
                            const IndexType numRows,
                            const IndexType numNonZerosPerRow,
                            const ValueType alpha,
                            const ValueType x[],
                            const IndexType numNonZeroRows,
                            const IndexType rowIndexes[],
                            const IndexType csrIA[],
                            const IndexType csrJA[],
                            const ValueType csrValues[],
                            class SyncToken* syncToken );

    static void setInterface( struct ELLUtilsInterface& ELLUtils );

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 csrIA[],
                            const IndexType csrJA[],
                            const ValueType csrValues[] );

    template <typename ValueType>
    static void sparseGEMV( ValueType result[],
                            const IndexType numRows,
                            const ValueType alpha,
                            const ValueType x[],
                            const IndexType numNonZeroRows,
                            const IndexType rowIndexes[],
                            const IndexType csrIA[],
                            const IndexType csrJA[],
                            const ValueType csrValues[] );

    LAMA_LOG_DECL_STATIC_LOGGER( logger );
};

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

}  // namespace lama

#endif  //  LAMA_ELL_STORAGE_UTILS_HPP_