Skip to content

Commit 2983520

Browse files
committed
fix hasattr(paddle.fluid.ir.PassDesc.OP, '__name__') error, test=develop
1 parent 70e6784 commit 2983520

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

python/paddle/fluid/ir.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -230,9 +230,6 @@ def __init__(self, type=None):
230230
self._type = type
231231

232232
def __getattr__(self, name):
233-
if self._type is not None:
234-
raise AttributeError(
235-
"type object 'OpHelper' has no attribute '{}'".format(name))
236233
op = PassDesc.OpHelper(name)
237234
op.Init()
238235
return op
@@ -261,7 +258,12 @@ def Init(self):
261258
self._op_idx = len(block.ops)
262259
self._op_desc = block.desc.append_op()
263260
self._op_desc.set_type(self._type)
264-
self._op_proto = OpProtoHolder.instance().get_op_proto(self._type)
261+
self._op_proto = OpProtoHolder.instance().op_proto_map.get(
262+
self._type)
263+
if self._op_proto is None:
264+
raise AttributeError(
265+
"type object 'OpHelper' has no attribute '{}'".format(
266+
self._type))
265267
block.ops.append(self)
266268

267269
def Attr(self, name):

0 commit comments

Comments
 (0)