common utils for cuda More...

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

error The error.

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

status The status.

const char* mxnet::common::cuda::CusolverGetErrorString ( cusolverStatus_t error )

inline

Get string representation of cuSOLVER errors.

Parameters

error The error.