From 4aaa2c618cb195e6d2de3cdbb37c1029296e75d7 Mon Sep 17 00:00:00 2001 From: lvliang Date: Tue, 1 Dec 2020 17:00:39 +0800 Subject: [PATCH] fix-bug-of-compile-failed-for-ge-pynative --- mindspore/ccsrc/pipeline/pynative/base.h | 1 + mindspore/ccsrc/pipeline/pynative/pynative_execute.cc | 1 + 2 files changed, 2 insertions(+) diff --git a/mindspore/ccsrc/pipeline/pynative/base.h b/mindspore/ccsrc/pipeline/pynative/base.h index 130ace395b..a89f06a75d 100644 --- 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 inputs_mask; bool is_dynamic_shape = false; std::string next_op_name = ""; diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index a7b3b00f45..adba9e7080 100644 --- 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; }