p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Always set tuple/list elements as used if it's the arguements of VirtualAbstractClosure, likely J's bprop func.

This commit is contained in:
Zhang Qinghua 2022-01-19 14:10:51 +08:00
parent 55b2a78b99
commit 5aab9d2e66
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/pipeline/jit/static_analysis/evaluator.cc
View File

@ -569,9 +569,17 @@ EvalResultPtr VirtualEvaluator::Eval(AnalysisEnginePtr, const AbstractBasePtrLis
MS_LOG(EXCEPTION) << "Arguments mismatch, parameters no: " << args_spec_list_.size() MS_LOG(EXCEPTION) << "Arguments mismatch, parameters no: " << args_spec_list_.size()
<< ", arguments no: " << args_spec_list.size(); << ", arguments no: " << args_spec_list.size();
} }
static const auto eliminate_unused_element = common::GetEnv("MS_DEV_ENABLE_DDE");
static const auto enable_eliminate_unused_element = (eliminate_unused_element != "0");
// Check each parameter and argument match; // Check each parameter and argument match;
for (std::size_t i = 0; i < args_spec_list.size(); i++) { for (std::size_t i = 0; i < args_spec_list.size(); i++) {
MS_EXCEPTION_IF_NULL(args_spec_list[i]); MS_EXCEPTION_IF_NULL(args_spec_list[i]);
// For VirtualAbstractClosure, likely J's bprop, we just set its tuple arguments as used before really grad.
if (enable_eliminate_unused_element && args_spec_list[i]->isa<abstract::AbstractSequence>()) {
MS_LOG(INFO) << "Notice: For VirtualAbstractClosure, update all use flags as true for arguments[" << i
<< "]: " << args_spec_list[i]->ToString();
SetSequenceElementsUseFlagsRecursively(args_spec_list[i], true);
}
(void)args_spec_list[i]->Join(args_spec_list_[i]); (void)args_spec_list[i]->Join(args_spec_list_[i]);
} }
return std::make_shared<EvalResult>(output_, std::make_shared<AttrValueMap>()); return std::make_shared<EvalResult>(output_, std::make_shared<AttrValueMap>());