提交 811eec9b authored 作者: James Bergstra's avatar James Bergstra 提交者: Amjad Almahairi

fix: local_log_add dont crash on >2 args

上级 030fbafb
...@@ -4186,6 +4186,11 @@ def local_log_add(node): ...@@ -4186,6 +4186,11 @@ def local_log_add(node):
z = node.inputs[0] z = node.inputs[0]
if z.owner and z.owner.op == T.add: if z.owner and z.owner.op == T.add:
zi = z.owner.inputs zi = z.owner.inputs
if len(zi) != 2:
# -- upgrading Maximum to handle multiple inputs wasn't trivial
# TODO
#raise NotImplementedError()
return
pre_exp = [x.owner.inputs[0] for x in zi pre_exp = [x.owner.inputs[0] for x in zi
if x.owner and x.owner.op == T.exp] if x.owner and x.owner.op == T.exp]
if len(pre_exp) == len(zi): if len(pre_exp) == len(zi):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论