utils.h

Go to the documentation of this file.

/*--------------------------------------------------------------------------
  Author/Modifier: Thomas Nowotny
  
  Institute: Center for Computational Neuroscience and Robotics
  University of Sussex
  Falmer, Brighton BN1 9QJ, UK 
  
  email to:  T.Nowotny@sussex.ac.uk
  
  initial version: 2010-02-07
   
  --------------------------------------------------------------------------*/

//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

#ifndef _UTILS_H_
#define _UTILS_H_ 

#include <cstdlib>
#include <iostream>
#include <string>

#ifndef CPU_ONLY
#include <cuda.h>
#include <cuda_runtime.h>
#endif

using namespace std;

// Forward declarations
class CodeStream;

#ifndef CPU_ONLY
//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

#if CUDA_VERSION >= 6050
#define CHECK_CU_ERRORS(call)                                        \
  {                                                                \
    CUresult error = call;                                        \
    if (error != CUDA_SUCCESS)                                        \
      {                                                                \
        const char *errStr;                                        \
        cuGetErrorName(error, &errStr);                                \
        cerr << __FILE__ << ": " <<  __LINE__;                        \
        cerr << ": cuda driver error " << error << ": ";        \
        cerr << errStr << endl;                                        \
        exit(EXIT_FAILURE);                                        \
      }                                                                \
  }
#else
#define CHECK_CU_ERRORS(call) call
#endif

// comment below and uncomment here when using CUDA that does not support cugetErrorName
//#define CHECK_CU_ERRORS(call) call
#define CHECK_CUDA_ERRORS(call)                                        \
  {                                                                \
    cudaError_t error = call;                                        \
    if (error != cudaSuccess)                                        \
      {                                                                \
        cerr << __FILE__ << ": " <<  __LINE__;                        \
        cerr << ": cuda runtime error " << error << ": ";        \
        cerr << cudaGetErrorString(error) << endl;                \
        exit(EXIT_FAILURE);                                        \
      }                                                                \
  }
#endif


//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

#define B(x,i) ((x) & (0x80000000 >> (i))) 

#define setB(x,i) x= ((x) | (0x80000000 >> (i))) 

#define delB(x,i) x= ((x) & (~(0x80000000 >> (i)))) 

//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

#define USE(expr) do { (void)(expr); } while (0) 

#ifndef CPU_ONLY
//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

CUresult cudaFuncGetAttributesDriver(cudaFuncAttributes *attr, CUfunction kern);
#endif


//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

inline void gennError(const string &error)
{
    cerr << "GeNN error: " << error << endl;
    exit(EXIT_FAILURE);
}

//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

size_t theSize(const string &type);


//--------------------------------------------------------------------------
//--------------------------------------------------------------------------

void writeHeader(CodeStream &os);


#endif  // _UTILS_H_