提交 e879bb63 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix opt crash with upcast/downcast

上级 d0e258dc
...@@ -4393,7 +4393,7 @@ def local_intdiv_by_one(node): ...@@ -4393,7 +4393,7 @@ def local_intdiv_by_one(node):
if node.op in [T.int_div]: if node.op in [T.int_div]:
if isinstance(node.inputs[1], T.TensorConstant) and \ if isinstance(node.inputs[1], T.TensorConstant) and \
numpy.all(node.inputs[1].value == 1): numpy.all(node.inputs[1].value == 1):
return [node.inputs[0]] return [node.inputs[0].astype(node.outputs[0].dtype)]
@gof.local_optimizer([T.pow]) @gof.local_optimizer([T.pow])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论