Browse Source

!9302 Fix bug of compile failed for ge pynative

From: @joylvliang
Reviewed-by: @chujinjin,@kisnwang
Signed-off-by: @chujinjin
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
b6ab253947
2 changed files with 2 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/pipeline/pynative/base.h
  2. +1
    -0
      mindspore/ccsrc/pipeline/pynative/pynative_execute.cc

+ 1
- 0
mindspore/ccsrc/pipeline/pynative/base.h View File

@@ -57,6 +57,7 @@ struct OpExecInfo {
AbstractBasePtr abstract;

py::list op_inputs;
py::dict op_attrs;
std::vector<bool> inputs_mask;
bool is_dynamic_shape = false;
std::string next_op_name = "";


+ 1
- 0
mindspore/ccsrc/pipeline/pynative/pynative_execute.cc View File

@@ -686,6 +686,7 @@ OpExecInfoPtr PynativeExecutor::GenerateOpExecInfo(const py::args &args) {
}
op_exec_info->prim_id = GetId(prim->GetPyObj());
op_exec_info->py_primitive = prim;
op_exec_info->op_attrs = py::getattr(args[PY_PRIM], "attrs");
op_exec_info->op_inputs = args[PY_INPUTS];
return op_exec_info;
}


Loading…
Cancel
Save