add_a_temp_solution_for_side_effect_grad_case_failed

pull/14525/head^2
lvliang 4 years ago
parent ea9c430564
commit c461d71de8

@ -504,6 +504,11 @@ PynativeAdjointPtr KPynativeCellImpl::ForgeMakeSequenceAdjoint(const CNodePtr &c
} else if (inp->isa<ValueNode>()) {
const auto &inp_value = GetValueNode(inp);
op_args.push_back(inp_value);
} else if (inp->isa<Parameter>()) {
auto param = inp->cast<ParameterPtr>();
const auto &abs = param->abstract();
MS_EXCEPTION_IF_NULL(abs);
op_args.push_back(abs->BuildValue());
} else {
MS_LOG(EXCEPTION) << "Input of MakeTuple/MakeLis is not a CNode or ValueNode, but: " << inp->DebugString();
}

@ -1782,7 +1782,7 @@ void GradExecutor::InitResourceAndDfBuilder(const std::string &cell_id, const py
auto new_param = curr_g_->add_parameter();
ValuePtr param_value = PyAttrValue(param);
MS_EXCEPTION_IF_NULL(param_value);
new_param->set_abstract(param_value->ToAbstract()->Broaden());
new_param->set_abstract(param_value->ToAbstract());
std::string param_id = GetId(param);
SetTupleArgsToGraphInfoMap(curr_g_, param, new_param, true);
SetNodeMapInGraphInfoMap(curr_g_, param_id, new_param);

Loading…
Cancel
Save