c10/cuda/CUDAException.h

#pragma once

#include "c10/util/Exception.h"
#include "cuda.h"

// Note [CHECK macro]
// ~~~~~~~~~~~~~~~~~~
// This is a macro so that AT_ERROR can get accurate __LINE__
// and __FILE__ information.  We could split this into a short
// macro and a function implementation if we pass along __LINE__
// and __FILE__, but no one has found this worth doing.

#define C10_CUDA_CHECK(EXPR)                                \
  do {                                                     \
    cudaError_t __err = EXPR;                              \
    if (__err != cudaSuccess) {                            \
      AT_ERROR("CUDA error: ", cudaGetErrorString(__err)); \
    }                                                      \
  } while (0)