Browse Source

fix-bug-pynative-gpu-memory-not-release

tags/v1.1.0
lvliang 5 years ago
parent
commit
087eb26ef5
1 changed files with 7 additions and 5 deletions
  1. +7
    -5
      mindspore/ccsrc/frontend/optimizer/irpass/special_op_eliminate.h

+ 7
- 5
mindspore/ccsrc/frontend/optimizer/irpass/special_op_eliminate.h View File

@@ -255,7 +255,7 @@ class PynativeEliminater : public OptimizerCaller {
MS_LOG(DEBUG) << "Start FillZero";
ValuePtr out = nullptr;
if (value->isa<Int32Imm>()) {
return value;
return MakeValue(value->cast<Int32ImmPtr>()->value());
}

if (value->isa<tensor::Tensor>()) {
@@ -298,9 +298,10 @@ class PynativeEliminater : public OptimizerCaller {
if (rep != nullptr) {
if (rep->isa<ValueNode>()) {
auto value_node = rep->cast<ValueNodePtr>();
value_node->set_value(FillZero(value_node->value()));
auto new_value_node = NewValueNode(FillZero(value_node->value()));
new_value_node->set_has_new_value(value_node->has_new_value());
MS_LOG(DEBUG) << "Zeros_like replace ok " << rep->DebugString(4);
return rep;
return new_value_node;
}
}
}
@@ -315,9 +316,10 @@ class PynativeEliminater : public OptimizerCaller {
if (rep != nullptr) {
if (rep->isa<ValueNode>()) {
auto value_node = rep->cast<ValueNodePtr>();
value_node->set_value(FillZero(value_node->value()));
auto new_value_node = NewValueNode(FillZero(value_node->value()));
new_value_node->set_has_new_value(value_node->has_new_value());
MS_LOG(DEBUG) << "Zeros_like replace ok 2 " << rep->DebugString(4);
return rep;
return new_value_node;
}
}
}


Loading…
Cancel
Save