xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix one stage core dump problem

This commit is contained in:
liangzhibo 2024-02-23 10:44:09 +08:00 committed by r1chardf1d0
parent 72ee43d8ad
commit 3a74cbbe83
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_build.cc
View File

@ -2861,6 +2861,9 @@ py::object GraphBuilder::ResolveCallable(CallNode *call_node, StopTraceReason *s
}
void GraphBuilder::ResolveClosure(const py::object &func_info, ValueNode *callable_node, FrameStates *frame) {
if (func_info.ptr() == nullptr) {
MS_LOG(INTERNAL_EXCEPTION) << "When resolving closure, get func_info failed.";
}
PyCodeObject *co = reinterpret_cast<PyCodeObject *>(PyFunction_GET_CODE(func_info.ptr()));
PyObject *closure = PyFunction_GET_CLOSURE(func_info.ptr());

4
mindspore/ccsrc/pipeline/pynative/grad/jit/jit_grad.cc
View File

@ -51,6 +51,10 @@ FrontendOpRunInfoPtr GetOpRunInfo(const py::object &out, const py::args &args, c
op_run_info->op_grad_info->input_abs.resize(op_run_info->input_size);
const auto &original_params = jit_forward_graph->parameters();
for (size_t i = 0; i < op_run_info->input_size; ++i) {
if (original_params.size() <= i) {
MS_LOG(INTERNAL_EXCEPTION) << "Index out of range for index: " << i
<< " and origin params size: " << original_params.size();
}
op_run_info->op_grad_info->input_abs[i] = original_params[i]->abstract();
}
if (modify_output) {