/home/brandes/workspace/LAMA/src/lama/cuda/CUDAELLUtils.hpp

Go to the documentation of this file.

#ifndef LAMA_CUDA_ELL_UTILS_HPP_
#define LAMA_CUDA_ELL_UTILS_HPP_

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

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT CUDAELLUtils
{
public:

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


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

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

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

    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 );

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

    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 normalGEMV( ValueType result[],
                            const ValueType alpha,
                            const ValueType x[],
                            const ValueType beta,
                            const ValueType y[],
                            const IndexType numRows,
                            const IndexType numNonZerosPerRows,
                            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 numNonZerosPerRows,
                            const ValueType alpha,
                            const ValueType x[],
                            const IndexType numNonZeroRows,
                            const IndexType rowIndexes[],
                            const IndexType csrIA[],
                            const IndexType csrJA[],
                            const ValueType csrValues[],
                            class SyncToken* syncToken );

    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 );

    static void setInterface( struct ELLUtilsInterface& ELLUtils );

private:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

};

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

}  // namespace lama

#endif  //  LAMA_CUDA_ELL_UTILS_HPP_