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

fix param

This commit is contained in:
r1chardf1d0 2024-02-28 11:17:20 +08:00
parent 3a74cbbe83
commit c0f4811b9a
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/pipeline/jit/pi/common.cc
View File

@ -934,6 +934,9 @@ std::vector<py::object> PackArgs(const PyFrameObject *frame) {
args[argi] = py::reinterpret_borrow<py::object>(PyCell_GET(cell));
}
}
if (vargs.ptr() != nullptr) {
PyList_Append(args.ptr(), vargs.ptr());
}
return {args, vargs, kwvargs};
}
@ -1140,9 +1143,6 @@ static py::object CallCompiledResults(PyThreadState *tstate, PyFrameObject *f, c
ValidateCompiledResults(c);
std::vector<py::object> packed_args = PackArgs(f);
if (packed_args[1].ptr() != nullptr) {
PyList_Append(packed_args[0].ptr(), packed_args[1].ptr());
}
py::object args = py::reinterpret_steal<py::object>(PyList_AsTuple(packed_args[0].ptr()));
py::object kwvargs = packed_args[2];

4
mindspore/ccsrc/pipeline/jit/pi/graph_capture/code_generator.cc
View File

@ -1578,7 +1578,9 @@ py::object MakeCodeFromCodeGen(const GraphBuilderPtr &builder, const GraphAnalyz
auto mind_builder = std::dynamic_pointer_cast<MindGraphBuilder>(builder);
auto mind_fg_builder = mind_builder->FGBuilder();
MS_EXCEPTION_IF_NULL(mind_fg_builder);
mind_fg_builder->EraseUnusedParameter();
if (analyzer->NeedInterpret()) {
mind_fg_builder->EraseUnusedParameter();
}
}
cg->Init(graph, &info);
cg->SetGlobals(py::cast<py::dict>(globals));