ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Classes | Macros | Functions
cuda.h File Reference
#include <cublas_v2.h>
#include <cuda_runtime.h>
#include <cusolverDn.h>
#include <thrust/complex.h>
Include dependency graph for cuda.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  GetTypeThrust< T >
 
struct  GetTypeThrust< std::complex< float > >
 
struct  GetTypeThrust< std::complex< double > >
 
struct  GetTypeCuda< T >
 
struct  GetTypeCuda< int >
 
struct  GetTypeCuda< float >
 
struct  GetTypeCuda< double >
 
struct  GetTypeCuda< int64_t >
 
struct  GetTypeCuda< std::complex< float > >
 
struct  GetTypeCuda< std::complex< double > >
 

Macros

#define THREADS_PER_BLOCK   256
 
#define cusolverErrcheck(res)
 
#define cublasErrcheck(res)
 
#define cudaErrcheck(res)
 
#define cudaCheckOnDebug()
 

Functions

void cusolverAssert (cusolverStatus_t code, const char *file, int line)
 
void cublasAssert (cublasStatus_t res, const char *file, int line)
 

Macro Definition Documentation

◆ cublasErrcheck

#define cublasErrcheck (   res)
Value:
{ \
cublasAssert((res), __FILE__, __LINE__); \
}

◆ cudaCheckOnDebug

#define cudaCheckOnDebug ( )

◆ cudaErrcheck

#define cudaErrcheck (   res)
Value:
{ \
if (res != cudaSuccess) \
{ \
fprintf(stderr, " Unexpected Device Error %s:%d: %s, %s\n", __FILE__, __LINE__, cudaGetErrorName(res), \
cudaGetErrorString(res)); \
exit(res); \
} \
}

◆ cusolverErrcheck

#define cusolverErrcheck (   res)
Value:
{ \
cusolverAssert((res), __FILE__, __LINE__); \
}

◆ THREADS_PER_BLOCK

#define THREADS_PER_BLOCK   256

Function Documentation

◆ cublasAssert()

void cublasAssert ( cublasStatus_t  res,
const char *  file,
int  line 
)
inline
Here is the call graph for this function:

◆ cusolverAssert()

void cusolverAssert ( cusolverStatus_t  code,
const char *  file,
int  line 
)
inline
Here is the call graph for this function: