Fix beam_search memory leak. (#10185)

2f53cd0a · gongweibao · GitHub · ee0497c4 · 2f53cd0a
隐藏空白更改
内联并排

Showing with 3 addition and 2 deletion

paddle/fluid/operators/beam_search_decode_op.h paddle/fluid/operators/beam_search_decode_op.h +3 -2

未找到文件。
--- a/paddle/fluid/operators/beam_search_decode_op.h
+++ b/paddle/fluid/operators/beam_search_decode_op.h
@@ -223,8 +223,9 @@ void BeamSearchDecoder<T>::ConvertSentenceVectorToLodTensor(
                               sentence_vector_list[src_idx].size());
  }
-  auto cpu_place = new paddle::platform::CPUPlace();
+  auto cpu_place = std::unique_ptr<paddle::platform::CPUPlace>(
-  paddle::platform::CPUDeviceContext cpu_ctx(*cpu_place);
+      new paddle::platform::CPUPlace());
+  paddle::platform::CPUDeviceContext cpu_ctx(*cpu_place.get());
  framework::LoD lod;
  lod.push_back(source_level_lod);