Browse Source

!2526 Fix memory leak in execution_tree.cc

Merge pull request !2526 from Alexey_Shevlyakov/alex_fix_memory_leak
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b4a66d47ef
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      mindspore/ccsrc/dataset/engine/execution_tree.cc

+ 3
- 3
mindspore/ccsrc/dataset/engine/execution_tree.cc View File

@@ -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));


Loading…
Cancel
Save