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

4 years ago · 4aaa2c618c
parent 7e1f2e2d61
commit 4aaa2c618c
2 changed files with 2 additions and 0 deletions
--- a/mindspore/ccsrc/pipeline/pynative/base.h
+++ b/mindspore/ccsrc/pipeline/pynative/base.h
@ -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 = "";
--- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
+++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc
@ -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;
 }