src/caffeine/common.hpp

   1 #ifndef CAFFEINE_COMMON_HPP_
   2 #define CAFFEINE_COMMON_HPP_
   3
   4 #include <iostream>
   5
   6 #include <boost/shared_ptr.hpp>
   7
   8 #include "driver_types.h"
   9
  10 namespace caffeine {
  11   using boost::shared_ptr;
  12 }
  13
  14 static std::ostream nullout(0);
  15
  16 // TODO(Yangqing): make a better logging scheme
  17 #define LOG_IF(condition) \
  18     !(condition) ? nullout : std::cout
  19
  20 #define CHECK(condition) \
  21     LOG_IF(condition) << "Check failed: " #condition " "
  22
  23 #ifndef NDEBUG
  24
  25 #define DCHECK(condition) CHECK(condition)
  26
  27 #else
  28
  29 #define DCHECK(condition)
  30
  31 #endif  // NDEBUG
  32
  33
  34 #define CUDA_CHECK(condition) \
  35     CUDA_LOG_IF(condition) << "Check failed: " #condition " "
  36
  37
  38 // TODO(Yangqing): make a better logging scheme
  39 #define CUDA_LOG_IF(condition) \
  40     ((condition) != cudaSuccess) ? nullout : std::cout
  41
  42 #define CUDA_CHECK(condition) \
  43     CUDA_LOG_IF(condition) << "Check failed: " #condition " "
  44
  45 #ifndef NDEBUG
  46
  47 #define CUDA_DCHECK(condition) CUDA_CHECK(condition)
  48
  49 #else
  50
  51 #define CUDA_DCHECK(condition)
  52
  53 #endif  // NDEBUG
  54
  55 #endif  // CAFFEINE_COMMON_HPP_