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

Go to the documentation of this file.

#ifndef LAMA_CUDA_JDS_UTILS_HPP_
#define LAMA_CUDA_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 CUDAJDSUtils
{
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 IndexType ilg2dlg( IndexType dlg[],
                              const IndexType numDiagonals,
                              const IndexType ilg[],
                              const IndexType numRows );

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

    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 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 solutionLocal[],
                            const IndexType numRows,
                            const ValueType diagonal[],
                            const IndexType ndlg_halo,
                            const IndexType jdsPermHalo[],
                            const IndexType jdsIlgHalo[],
                            const IndexType jdsDlgHalo[],
                            const IndexType jdsJAHalo[],
                            const ValueType jdsValuesHalo[],
                            const ValueType oldSolutionHalo[],
                            const ValueType omega,
                            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 perm[],
                            const IndexType jdsILG[],
                            const IndexType ndlg,
                            const IndexType jdsDLG[],
                            const IndexType jdsJA[],
                            const ValueType jdsValues[],
                            class SyncToken* syncToken );

    static void setInterface( struct JDSUtilsInterface& JDSUtils );

private:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

};

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

}  // namespace lama

#endif //  LAMA_CUDA_JDS_UTILS_HPP_