first try
[jacinto-ai/caffe-jacinto.git] / src / caffeine / common.hpp
1 #ifndef CAFFEINE_COMMON_HPP_
2 #define CAFFEINE_COMMON_HPP_
4 #include <iostream>
6 #include <boost/shared_ptr.hpp>
8 #include "driver_types.h"
10 namespace caffeine {
11   using boost::shared_ptr;
12 }
14 static std::ostream nullout(0);
16 // TODO(Yangqing): make a better logging scheme
17 #define LOG_IF(condition) \
18     ((condition) != cudaSuccess) ? nullout : std::cout
20 #define CUDA_CHECK(condition) \
21     LOG_IF(condition) << "Check failed: " #condition " "
22     
23 #endif  // CAFFEINE_COMMON_HPP_