have rng_stream initialize RNG if not already initialized

diff --git a/include/caffe/common.hpp b/include/caffe/common.hpp
index 1829026..af7dcc4 100644 (file)
--- a/include/caffe/common.hpp
+++ b/include/caffe/common.hpp
@@ -96,6 +96,9 @@ class Caffe {
 
   // Getters for boost rng, curand, and cublas handles
   inline static const RNG& rng_stream() {
+    if (!Get().random_generator_) {
+      Get().random_generator_.reset(new RNG());
+    }
     return *(Get().random_generator_);
   }
   inline static cublasHandle_t cublas_handle() { return Get().cublas_handle_; }