mxnet
|
CUDA debugging utilities. More...
#include <dmlc/logging.h>
#include <cuda_runtime.h>
#include <cublas_v2.h>
#include <curand.h>
Go to the source code of this file.
Namespaces | |
mxnet | |
namespace of mxnet | |
mxnet::common | |
mxnet::common::cuda | |
common utils for cuda | |
Macros | |
#define | CHECK_CUDA_ERROR(msg) |
Check CUDA error. More... | |
#define | CUDA_CALL(func) |
Protected CUDA call. More... | |
#define | CUBLAS_CALL(func) |
Protected cuBLAS call. More... | |
#define | CURAND_CALL(func) |
Protected cuRAND call. More... | |
Functions | |
const char * | mxnet::common::cuda::CublasGetErrorString (cublasStatus_t error) |
Get string representation of cuBLAS errors. More... | |
const char * | mxnet::common::cuda::CurandGetErrorString (curandStatus_t status) |
Get string representation of cuRAND errors. More... | |
CUDA debugging utilities.
Copyright (c) 2015 by Contributors
#define CHECK_CUDA_ERROR | ( | msg | ) |
Check CUDA error.
msg | Message to print if an error occured. |
#define CUBLAS_CALL | ( | func | ) |
Protected cuBLAS call.
func | Expression to call. |
It checks for cuBLAS errors after invocation of the expression.
#define CUDA_CALL | ( | func | ) |
Protected CUDA call.
func | Expression to call. |
It checks for CUDA errors after invocation of the expression.
#define CURAND_CALL | ( | func | ) |
Protected cuRAND call.
func | Expression to call. |
It checks for cuRAND errors after invocation of the expression.