Fix compile

8 years ago · 42062c38b1
parent 8b877dd7c8
commit 42062c38b1
1 changed files with 2 additions and 2 deletions
--- a/paddle/operators/beam_search_decode_op.cc
+++ b/paddle/operators/beam_search_decode_op.cc
@ -57,8 +57,8 @@ class BeamSearchDecodeOp : public framework::OperatorBase {
      : OperatorBase(type, inputs, outputs, attrs) {}
  void Run(const framework::Scope& scope,
           const platform::Place& dev_place) const override {
-    platform::DeviceContextPool& pool = platform::DeviceContextPool::Get();
-    auto& dev_ctx = *pool.Borrow(dev_place);
+    platform::DeviceContextPool& pool = platform::DeviceContextPool::Instance();
+    auto& dev_ctx = *pool.Get(dev_place);

    framework::ExecutionContext ctx(*this, scope, dev_ctx);