common: added DeviceQuery() function

[jacinto-ai/caffe-jacinto.git] / include / caffe / common.hpp

diff --git a/include/caffe/common.hpp b/include/caffe/common.hpp
index cc6e31b705b0b22840d6e51821e0489eee924619..7362b233780d7d205906df1e53bbab2950e86fa5 100644 (file)
--- a/include/caffe/common.hpp
+++ b/include/caffe/common.hpp
@@ -91,6 +91,8 @@ class Caffe {
   inline static void set_phase(Phase phase) { Get().phase_ = phase; }
   // Sets the random seed of both MKL and curand
   static void set_random_seed(const unsigned int seed);
+  // Prints the current GPU status.
+  static void DeviceQuery();
 
  protected:
   cublasHandle_t cublas_handle_;