niftkCUDAUtils.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/

#ifndef niftkCUDAUtils_h
#define niftkCUDAUtils_h

#include <niftkCUDAKernelsWin32ExportHeader.h>
#include <cuda.h>
#include <cstdio>

namespace niftk
{

static void HandleCUDAError( cudaError_t err,
                             const char *file,
                             int line ) {
  if (err != cudaSuccess) {
    printf( "HandleCUDAError: Error=%i, message=%s in %s at line %d\n", err, cudaGetErrorString( err ), file, line );
    exit( EXIT_FAILURE );
  }
}

#define niftkCUDACall( err ) ( HandleCUDAError( err, __FILE__, __LINE__ ))

} // end namespace

#endif