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

Go to the documentation of this file.

#ifndef LAMA_CUDABLAS2_HPP_
#define LAMA_CUDABLAS2_HPP_

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

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

#include <lama/openmp/BLASHelper.hpp>

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

#include <cublas.h>
#include <cuda_runtime_api.h>

namespace lama
{

class LAMA_DLL_IMPORTEXPORT CUDABLAS2
{
public:

    template<typename T>
    static void gemv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_TRANSPOSE trans,
        const IndexType m,
        const IndexType n,
        const T alpha,
        const T* A,
        const IndexType lda,
        const T* x,
        const IndexType incX,
        const T beta,
        T* y,
        const IndexType incY,
        SyncToken* syncToken );

    template<typename T>
    static void symv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* A,
        const IndexType lda,
        const T* x,
        const IndexType incX,
        const T beta,
        T* y,
        const IndexType incY,
        SyncToken* syncToken );

    template<typename T>
    static void trmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const T* A,
        const IndexType lda,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

    template<typename T>
    static void trsv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const T* A,
        const IndexType lda,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

    template<typename T>
    static void gbmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_TRANSPOSE trans,
        const IndexType m,
        const IndexType n,
        const IndexType kl,
        const IndexType ku,
        const T alpha,
        const T* A,
        const IndexType lda,
        const T* x,
        const IndexType incX,
        const T beta,
        T* y,
        const IndexType incY,
        SyncToken* syncToken );

    template<typename T>
    static void sbmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const IndexType k,
        const T alpha,
        const T* A,
        const IndexType lda,
        const T* x,
        const IndexType incX,
        const T beta,
        T* y,
        const IndexType incY,
        SyncToken* syncToken );

    template<typename T>
    static void tbmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const IndexType k,
        const T* A,
        const IndexType lda,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

    template<typename T>
    static void tbsv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const IndexType k,
        const T* A,
        const IndexType lda,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

    template<typename T>
    static void ger(
        const enum CBLAS_ORDER order,
        const IndexType m,
        const IndexType n,
        const T alpha,
        const T* x,
        const IndexType incX,
        const T* y,
        const IndexType incY,
        T* A,
        const IndexType lda,
        SyncToken* syncToken );

    template<typename T>
    static void syr(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* x,
        const IndexType incX,
        T* A,
        const IndexType lda,
        SyncToken* syncToken );

    template<typename T>
    static void syr2(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* x,
        const IndexType incX,
        const T* y,
        const IndexType incY,
        T* A,
        const IndexType lda,
        SyncToken* syncToken );

    template<typename T>
    static void spmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* AP,
        const T* x,
        const IndexType incX,
        const T beta,
        T* y,
        const IndexType incY,
        SyncToken* syncToken );

    template<typename T>
    static void spr(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* x,
        const IndexType incX,
        T* AP,
        SyncToken* syncToken );

    template<typename T>
    static void spr2(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const IndexType n,
        const T alpha,
        const T* x,
        const IndexType incX,
        const T* y,
        const IndexType incY,
        T* AP,
        SyncToken* syncToken );

    template<typename T>
    static void tpmv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const T* AP,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

    template<typename T>
    static void tpsv(
        const enum CBLAS_ORDER order,
        const enum CBLAS_UPLO uplo,
        const enum CBLAS_TRANSPOSE trans,
        const enum CBLAS_DIAG diag,
        const IndexType n,
        const T* Ap,
        T* x,
        const IndexType incX,
        SyncToken* syncToken );

private:

    LAMA_LOG_DECL_STATIC_LOGGER( logger );

};

} /* namespace lama */

#endif // LAMA_CUDABLAS2_HPP_