"switch to shared_ptr"

PaddlePaddle · Jul 11, 2017 · b871641 · b871641
1 parent 18e65b0
commit b871641
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paddle/framework/net.cc b/paddle/framework/net.cc
@@ -11,7 +11,7 @@ void PlainNet::InferShape(Scope* scope) {
   }
 }
 
-void PlainNet::Run(Scope* scope, DeviceContext* ctx) {
+void PlainNet::Run(std::shared_ptr<Scope> scope, DeviceContext* ctx) {
   for (auto& op : ops_) {
     op.Run(ctx);
   }

diff --git a/paddle/framework/net.h b/paddle/framework/net.h
@@ -69,7 +69,7 @@ class Net {
    * environment for ops. `begin` and `end` specify the scope of `ops_` to run,
    * If no positive indexes are provided, all operators in `ops_` will run.
    */
-  virtual void Run(Scope *scope, DeviceContext *ctx) = 0;
+  virtual void Run(std::shared_ptr<Scope> scope, DeviceContext *ctx) = 0;
 
   /**
    * @brief Add an Operator according to `def`.
@@ -123,7 +123,7 @@ class PlainNet : public Net {
    * scope will be used instead. If no OpContext is provicded, default context
    * will be used.
    */
-  virtual void Run(Scope *scope, DeviceContext *ctx) override;
+  virtual void Run(std::shared_ptr<Scope> scope, DeviceContext *ctx) override;
 
   /**
    * @brief Add an operator to this network.