Browse Source

fix-bug-of-compile-failed-for-ge-pynative

tags/v1.1.0
lvliang 5 years ago
parent
commit
4aaa2c618c
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