X-Git-Url: https://git.ti.com/gitweb?p=jacinto-ai%2Fcaffe-jacinto.git;a=blobdiff_plain;f=include%2Fcaffe%2Fcommon.hpp;h=7362b233780d7d205906df1e53bbab2950e86fa5;hp=cc6e31b705b0b22840d6e51821e0489eee924619;hb=4bfe30ce07ab3b291f65273599f090a5b4d7a114;hpb=7a0659875fb9af664da1fb9a828412e709fd74ae

diff --git a/include/caffe/common.hpp b/include/caffe/common.hpp
index cc6e31b7..7362b233 100644
--- 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_;