cuBLAS.hpp File Reference

Include dependency graph for cuBLAS.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	qmcplusplus
	helper functions for EinsplineSetBuilder
	qmcplusplus::cuBLAS
	interface to cuBLAS calls for different data types S/C/D/Z

Macros
#define	castNativeType   castCUDAType
#define	cublasErrorCheck(ans, cause)

Functions
void	cublasAssert (cublasStatus_t code, const std::string &cause, const char *file, int line, bool abort=true)
	prints cuBLAS error messages. Always use cublasErrorCheck macro. More...
cublasOperation_t	convertOperation (const char trans)
cublasStatus_t	geam (cublasHandle_t &handle, cublasOperation_t &transa, cublasOperation_t &transb, int m, int n, const float *alpha, const float *A, int lda, const float *beta, const float *B, int ldb, float *C, int ldc)
cublasStatus_t	geam (cublasHandle_t &handle, cublasOperation_t transa, cublasOperation_t transb, int m, int n, const double *alpha, const double *A, int lda, const double *beta, const double *B, int ldb, double *C, int ldc)
cublasStatus_t	geam (cublasHandle_t &handle, cublasOperation_t transa, cublasOperation_t transb, int m, int n, const std::complex< double > *alpha, const std::complex< double > *A, int lda, const std::complex< double > *beta, const std::complex< double > *B, int ldb, std::complex< double > *C, int ldc)
cublasStatus_t	geam (cublasHandle_t &handle, cublasOperation_t transa, cublasOperation_t transb, int m, int n, const std::complex< float > *alpha, const std::complex< float > *A, int lda, const std::complex< float > *beta, const std::complex< float > *B, int ldb, std::complex< float > *C, int ldc)
cublasStatus_t	getrf_batched (cublasHandle_t &handle, int n, float *A[], int lda, int *PivotArray, int *infoArray, int batchSize)
cublasStatus_t	getrf_batched (cublasHandle_t &handle, int n, double *A[], int lda, int *PivotArray, int *infoArray, int batchSize)
cublasStatus_t	getrf_batched (cublasHandle_t &handle, int n, std::complex< float > *A[], int lda, int *PivotArray, int *infoArray, int batchSize)
cublasStatus_t	getrf_batched (cublasHandle_t &handle, int n, std::complex< double > *A[], int lda, int *PivotArray, int *infoArray, int batchSize)
cublasStatus_t	getri_batched (cublasHandle_t &handle, int n, float *A[], int lda, int *PivotArray, float *C[], int ldc, int *infoArray, int batchSize)
cublasStatus_t	getri_batched (cublasHandle_t &handle, int n, double *A[], int lda, int *PivotArray, double *C[], int ldc, int *infoArray, int batchSize)
cublasStatus_t	getri_batched (cublasHandle_t &handle, int n, std::complex< float > *A[], int lda, int *PivotArray, std::complex< float > *C[], int ldc, int *infoArray, int batchSize)
cublasStatus_t	getri_batched (cublasHandle_t &handle, int n, std::complex< double > *A[], int lda, int *PivotArray, std::complex< double > *C[], int ldc, int *infoArray, int batchSize)

Macro Definition Documentation

castNativeType

#define castNativeType   castCUDAType

Definition at line 23 of file cuBLAS.hpp.

Referenced by qmcplusplus::cuBLAS::geam().

cublasErrorCheck

#define cublasErrorCheck	(		ans,
			cause
	)

Value:

{                                                 \
    cublasAssert((ans), cause, __FILE__, __LINE__); \
  }

Definition at line 34 of file cuBLAS.hpp.

Referenced by BLASHandle< PlatformKind::CUDA >::BLASHandle(), CUDAHandles::CUDAHandles(), DiracMatrixComputeCUDA< VALUE_FP >::DiracMatrixComputeCUDA(), qmcplusplus::compute::BLAS::gemm(), qmcplusplus::compute::BLAS::gemm_batched(), qmcplusplus::compute::BLAS::gemv(), qmcplusplus::compute::BLAS::ger(), DiracMatrixComputeCUDA< VALUE_FP >::invert_transpose(), DiracMatrixComputeCUDA< VALUE_FP >::mw_computeInvertAndLog(), DiracMatrixComputeCUDA< VALUE_FP >::mw_invertTranspose(), qmcplusplus::TEST_CASE(), BLASHandle< PlatformKind::CUDA >::~BLASHandle(), CUDAHandles::~CUDAHandles(), and DiracMatrixComputeCUDA< VALUE_FP >::~DiracMatrixComputeCUDA().

Function Documentation

cublasAssert()

void cublasAssert ( cublasStatus_t  code, const std::string &  cause, const char *  file, int  line, bool  abort = true  )

inline

prints cuBLAS error messages. Always use cublasErrorCheck macro.

Definition at line 40 of file cuBLAS.hpp.

References CUBLAS_STATUS_ALLOC_FAILED, CUBLAS_STATUS_ARCH_MISMATCH, CUBLAS_STATUS_EXECUTION_FAILED, CUBLAS_STATUS_INTERNAL_ERROR, CUBLAS_STATUS_INVALID_VALUE, CUBLAS_STATUS_MAPPING_ERROR, CUBLAS_STATUS_NOT_INITIALIZED, CUBLAS_STATUS_NOT_SUPPORTED, and CUBLAS_STATUS_SUCCESS.

{
  if (code != CUBLAS_STATUS_SUCCESS)
  {
    std::string cublas_error;
    switch (code)
    {
    case CUBLAS_STATUS_NOT_INITIALIZED:
      cublas_error = "CUBLAS_STATUS_NOT_INITIALIZED";
      break;
    case CUBLAS_STATUS_ALLOC_FAILED:
      cublas_error = "CUBLAS_STATUS_ALLOC_FAILED";
      break;
    case CUBLAS_STATUS_INVALID_VALUE:
      cublas_error = "CUBLAS_STATUS_INVALID_VALUE";
      break;
    case CUBLAS_STATUS_ARCH_MISMATCH:
      cublas_error = "CUBLAS_STATUS_ARCH_MISMATCH";
      break;
    case CUBLAS_STATUS_MAPPING_ERROR:
      cublas_error = "CUBLAS_STATUS_MAPPING_ERROR";
      break;
    case CUBLAS_STATUS_EXECUTION_FAILED:
      cublas_error = "CUBLAS_STATUS_EXECUTION_FAILED";
      break;
    case CUBLAS_STATUS_INTERNAL_ERROR:
      cublas_error = "CUBLAS_STATUS_INTERNAL_ERROR";
      break;
    case CUBLAS_STATUS_NOT_SUPPORTED:
      cublas_error = "CUBLAS_STATUS_NOT_SUPPORTED";
      break;
#ifndef QMC_CUDA2HIP
    case CUBLAS_STATUS_LICENSE_ERROR:
      cublas_error = "CUBLAS_STATUS_LICENSE_ERROR";
      break;
#endif
    default:
      cublas_error = "<unknown>";
    }

    std::ostringstream err;
    err << "cublasAssert: " << cublas_error << ", file " << file << " , line " << line << std::endl
        << cause << std::endl;
    std::cerr << err.str();
    //if (abort) exit(code);
    throw std::runtime_error(cause);
  }
}