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

fix compile error

This commit is contained in:
r1chardf1d0 2024-03-02 10:46:43 +08:00
parent 6ae255955b
commit 8449dff594
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2603,7 +2603,7 @@ py::object MindGraphBuilder::FGAddNode(CallNode *call_node, const py::object &ca
return py::object();
}
std::vector<py::object> MindGraphBuilder::GetNewArgs(CallNode *call_node, AObj *vobj) {
std::vector<py::object> MindGraphBuilder::GetNewArgs(CallNode *call_node, AObject *vobj) {
std::vector<py::object> new_args;
vobj = vobj ? vobj : call_node->GetVobj();
if (vobj->GetType() == AObject::kTypeCFunction) {

2
mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_build.h
View File

@ -335,7 +335,7 @@ class MindGraphBuilder : public GraphBuilder {
bool DoBuildOp(const Instr &instr) override;
private:
std::vector<py::object> GetNewArgs(CallNode *call_node, AObj *vobj = nullptr);
std::vector<py::object> GetNewArgs(CallNode *call_node, AObject *vobj = nullptr);
bool IsFuncInWhiteList(const py::object &f, std::string *special_func_key);
bool HandleFuncInWhiteList(const std::string &key, CallNode *n) override;