make rng_ a private member of Generator

diff --git a/src/caffe/common.cpp b/src/caffe/common.cpp
index 65c6800..1ea329f 100644 (file)
--- a/src/caffe/common.cpp
+++ b/src/caffe/common.cpp
@@ -125,6 +125,8 @@ class Caffe::RNG::Generator {
   explicit Generator(unsigned int seed) : rng_(new caffe::rng_t(seed)) {}
   explicit Generator(const caffe::rng_t& other) :
       rng_(new caffe::rng_t(other)) {}
+  const caffe::rng_t& rng() const { return *rng_; }
+ private:
   shared_ptr<caffe::rng_t> rng_;
 };
 
@@ -141,7 +143,7 @@ Caffe::RNG& Caffe::RNG::operator=(const RNG& other) {
 }
 
 const void* Caffe::RNG::generator() const {
-  return static_cast<const void*>(generator_->rng_.get());
+  return static_cast<const void*>(&generator_->rng());
 }
 
 void Caffe::RNG::set_generator(const void* other_rng) {