vt::cuda::CuBLAS#

inline void vt::cuda::check_cublas_status(cublasStatus_t status, const std::string &message)#

check the status of the CUBLAS library.

Parameters:

  • status – The status of the CUBLAS library.

  • message – The message to be displayed if the status is not CUBLAS_STATUS_SUCCESS.

Throws:

std::runtime_error – If the status is not CUBLAS_STATUS_SUCCESS.

struct CuBLASHandleDeleter#

Deleter for the CuBLAS handle.

inline CuBLASHandleT vt::cuda::create_cublas_handle()#

Create a CuBLAS unique pointer handle.

Returns:

CuBLASHandleT: The CuBLAS unique pointer handle.

class CuBLAS#

Public Functions

inline cublasHandle_t get_handle() const#

Get the CuBLAS handle.

Returns:

cublasHandle_t: The CuBLAS handle.

Public Static Functions

static inline CuBLAS &get_instance()#

Get the Instance object. It returns the singleton instance of the CuBLAS handle.

Returns:

CuBLAS&

static CuBLAS &vt::cuda::cublas = CuBLAS::get_instance()#
template<typename T>
struct CuBLASFuncType#

The type of the CuBLAS functions.

Template Parameters:

T – The data type of the CuBLAS functions.

template<typename T>
struct CuBLASFunc#

The CuBLAS functions for different data types.

Template Parameters:

T – The data type of the CuBLAS functions.