QMCPACK
BLASHandle< PlatformKind::CUDA > Class Template Reference
+ Collaboration diagram for BLASHandle< PlatformKind::CUDA >:

Public Member Functions

 BLASHandle (Queue< PlatformKind::CUDA > &queue)
 
 ~BLASHandle ()
 

Public Attributes

const cudaStream_t h_stream
 
cublasHandle_t h_cublas
 

Detailed Description

template<>
class qmcplusplus::compute::BLASHandle< PlatformKind::CUDA >

Definition at line 30 of file AccelBLAS_CUDA.hpp.

Constructor & Destructor Documentation

◆ BLASHandle()

BLASHandle ( Queue< PlatformKind::CUDA > &  queue)
inline

Definition at line 38 of file AccelBLAS_CUDA.hpp.

References cublasCreate, cublasErrorCheck, cublasSetStream, and qmcplusplus::queue.

38  : h_stream(queue.getNative())
39  {
40  cublasErrorCheck(cublasCreate(&h_cublas), "cublasCreate failed!");
41  cublasErrorCheck(cublasSetStream(h_cublas, h_stream), "cublasSetStream failed!");
42  }
#define cublasCreate
Definition: cuda2hip.h:37
#define cublasSetStream
Definition: cuda2hip.h:39
#define cublasErrorCheck(ans, cause)
Definition: cuBLAS.hpp:34

◆ ~BLASHandle()

~BLASHandle ( )
inline

Definition at line 44 of file AccelBLAS_CUDA.hpp.

References cublasDestroy, and cublasErrorCheck.

44 { cublasErrorCheck(cublasDestroy(h_cublas), "cublasDestroy failed!"); }
#define cublasDestroy
Definition: cuda2hip.h:38
#define cublasErrorCheck(ans, cause)
Definition: cuBLAS.hpp:34

Member Data Documentation

◆ h_cublas

◆ h_stream


The documentation for this class was generated from the following file: