solver restructuring: now all prototxt are specified in the solver protocol buffer

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

diff --git a/include/caffe/common.hpp b/include/caffe/common.hpp
index 485a86947f5f9834b056593dd60ca6507b3f935e..af42772bac43951a2861892e1dfd5d4b57ce355e 100644 (file)
--- a/include/caffe/common.hpp
+++ b/include/caffe/common.hpp
@@ -94,6 +94,9 @@ 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);
+  // Sets the device. Since we have cublas and curand stuff, set device also
+  // requires us to reset those values.
+  static void SetDevice(const int device_id);
   // Prints the current GPU status.
   static void DeviceQuery();