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

Go to the documentation of this file.

#ifndef LAMA_CUDA_CSR_UTILS_HPP_
#define LAMA_CUDA_CSR_UTILS_HPP_

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

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

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT CUDACSRUtils
{
public:

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

    static void offsets2sizes( IndexType sizes[], const IndexType offsets[], const IndexType n );

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

    template <typename ValueType>
    static void convertCSR2CSC ( IndexType cscIA[],
                                 IndexType cscJA[],
                                 ValueType cscValues[],
                                 const IndexType csrIA[],
                                 const IndexType csrJA[],
                                 const ValueType csrValues[],
                                 int numRows, int numColumns, int numValues );

    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[],
                            class SyncToken* syncToken );

    template <typename ValueType>
    static void sparseGEMV( ValueType result[],
                            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* const solution,
           const IndexType csrIA[],
           const IndexType csrJA[],
           const ValueType csrValues[],
           const ValueType rhs[],
           const ValueType oldSolution[],
           const ValueType omega,
           const IndexType numRows,
           class SyncToken* syncToken );

    template<typename ValueType>
    static void jacobiHalo(
            ValueType solution[],
            const IndexType localIA[],
            const ValueType localValues[],
            const IndexType haloIA[],
            const IndexType haloJA[],
            const ValueType haloValues[],
            const IndexType haloRowIndexes[],
            const ValueType oldSolution[],
            const ValueType omega,
            const IndexType numNonEmptyRows );

static void setInterface( struct CSRUtilsInterface& CSRUtils );

private:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

};

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

}  // namespace lama

#endif  //  LAMA_CUDA_CSR_UTILS_HPP_