提交 9da6df1a authored 作者: Hengjean's avatar Hengjean

Fixed error in CAReduce test

上级 4b63cf4a
...@@ -308,7 +308,10 @@ def get_nothing(r, name, sub): ...@@ -308,7 +308,10 @@ def get_nothing(r, name, sub):
def get_c_declare(r, name, sub, check_input=True): def get_c_declare(r, name, sub, check_input=True):
"""Wrapper around c_declare that declares py_name""" """Wrapper around c_declare that declares py_name"""
if any([c == 'output' or getattr(c.op, 'check_input', True) for (c, _) in r.clients]): if any([c == 'output' or getattr(c.op, 'check_input', True) for (c, _)
in r.clients]) or r.owner and getattr(r.owner.op,
'check_input', True):
c_declare = r.type.c_declare(name, sub, True) c_declare = r.type.c_declare(name, sub, True)
else: else:
c_declare = r.type.c_declare(name, sub, False) c_declare = r.type.c_declare(name, sub, False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论