mxnet
Classes | Functions
mxnet::common::cuda Namespace Reference

common utils for cuda More...

Classes

class  DeviceStore
 

Functions

const char * CublasGetErrorString (cublasStatus_t error)
 Get string representation of cuBLAS errors. More...
 
const char * CusolverGetErrorString (cusolverStatus_t error)
 Get string representation of cuSOLVER errors. More...
 
const char * CurandGetErrorString (curandStatus_t status)
 Get string representation of cuRAND errors. More...
 
template<typename DType >
DType __device__ CudaMax (DType a, DType b)
 
template<typename DType >
DType __device__ CudaMin (DType a, DType b)
 

Detailed Description

common utils for cuda

Function Documentation

const char* mxnet::common::cuda::CublasGetErrorString ( cublasStatus_t  error)
inline

Get string representation of cuBLAS errors.

Parameters
errorThe error.
Returns
String representation.
template<typename DType >
DType __device__ mxnet::common::cuda::CudaMax ( DType  a,
DType  b 
)
inline
template<typename DType >
DType __device__ mxnet::common::cuda::CudaMin ( DType  a,
DType  b 
)
inline
const char* mxnet::common::cuda::CurandGetErrorString ( curandStatus_t  status)
inline

Get string representation of cuRAND errors.

Parameters
statusThe status.
Returns
String representation.
const char* mxnet::common::cuda::CusolverGetErrorString ( cusolverStatus_t  error)
inline

Get string representation of cuSOLVER errors.

Parameters
errorThe error.
Returns
String representation.