fix memory leak

5 years ago · 830af41c11
parent cdaaf38206
commit 830af41c11
1 changed files with 3 additions and 3 deletions
--- a/mindspore/ccsrc/dataset/engine/execution_tree.cc
+++ b/mindspore/ccsrc/dataset/engine/execution_tree.cc
@ -209,10 +209,10 @@ Status ExecutionTree::Prepare() {

 Status ExecutionTree::PrepareTreePreAction() {
  bool modified = false;
-  std::vector<Pass *> pre_actions;
+  std::vector<std::unique_ptr<Pass>> pre_actions;
  // Construct pre actions
-  pre_actions.push_back(new MapColumnReorder());
-  pre_actions.push_back(new GlobalShufflePass());
+  pre_actions.push_back(std::make_unique<MapColumnReorder>());
+  pre_actions.push_back(std::make_unique<GlobalShufflePass>());
  // Apply pre action passes
  for (auto &pass : pre_actions) {
    RETURN_IF_NOT_OK(pass->Run(this, &modified));