cudaUtility.h

/*
 * inference-101
 */
 
#ifndef __CUDA_UTILITY_H_
#define __CUDA_UTILITY_H_
 
 
#include <cuda_runtime.h>
#include <cuda.h>
#include <stdio.h>
#include <string.h>
 
 
#define CUDA(x)             cudaCheckError((x), #x, __FILE__, __LINE__)
 
#define CUDA_SUCCESS(x)         (CUDA(x) == cudaSuccess)
 
#define CUDA_FAILED(x)          (CUDA(x) != cudaSuccess)
 
#define CUDA_VERIFY(x)          if(CUDA_FAILED(x))  return false;
 
#define LOG_CUDA "[cuda]   "
 
/*
 * define this if you want all cuda calls to be printed
 */
//#define CUDA_TRACE
 
 
 
inline cudaError_t cudaCheckError(cudaError_t retval, const char* txt, const char* file, int line )
{
#if !defined(CUDA_TRACE)
    if( retval == cudaSuccess)
        return cudaSuccess;
#endif
 
    //int activeDevice = -1;
    //cudaGetDevice(&activeDevice);
 
    //Log("[cuda]   device %i  -  %s\n", activeDevice, txt);
    
    printf(LOG_CUDA "%s\n", txt);
 
 
    if( retval != cudaSuccess )
    {
        printf(LOG_CUDA "   %s (error %u) (hex 0x%02X)\n", cudaGetErrorString(retval), retval, retval);
        printf(LOG_CUDA "   %s:%i\n", file, line);  
    }
 
    return retval;
}
 
 
inline __device__ __host__ int iDivUp( int a, int b )       { return (a % b != 0) ? (a / b + 1) : (a / b); }
 
 
 
#endif