Skip to content

Commit 2f53cd0

Browse files
authored
Fix beam_search memory leak. (#10185)
1 parent ee0497c commit 2f53cd0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

paddle/fluid/operators/beam_search_decode_op.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,8 +223,9 @@ void BeamSearchDecoder<T>::ConvertSentenceVectorToLodTensor(
223223
sentence_vector_list[src_idx].size());
224224
}
225225

226-
auto cpu_place = new paddle::platform::CPUPlace();
227-
paddle::platform::CPUDeviceContext cpu_ctx(*cpu_place);
226+
auto cpu_place = std::unique_ptr<paddle::platform::CPUPlace>(
227+
new paddle::platform::CPUPlace());
228+
paddle::platform::CPUDeviceContext cpu_ctx(*cpu_place.get());
228229

229230
framework::LoD lod;
230231
lod.push_back(source_level_lod);

0 commit comments

Comments
 (0)