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

fix the printing of warning message.

上级 f8f10efe
...@@ -1281,7 +1281,7 @@ def local_elemwise_fusion(node): ...@@ -1281,7 +1281,7 @@ def local_elemwise_fusion(node):
except NotImplementedError: except NotImplementedError:
catch = True catch = True
if catch: if catch:
_logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion." % str(i.owner.op.scalar_op)) _logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion of this op." % str(i.owner.op.scalar_op))
do_fusion=False do_fusion=False
if do_fusion: if do_fusion:
...@@ -1307,10 +1307,10 @@ def local_elemwise_fusion(node): ...@@ -1307,10 +1307,10 @@ def local_elemwise_fusion(node):
["x" for x in s_g], ["x" for x in s_g],
"z",{}) "z",{})
except MethodNotDefined: except MethodNotDefined:
_logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion." % str(i.owner.op.scalar_op)) _logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion of this op." % str(s_new_out.owner.op))
return False return False
except NotImplementedError: except NotImplementedError:
_logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion." % str(s_new_out.op.scalar_op)) _logger.info("%s does not implement the c_code function. As well as being potentially slow, this disables loop fusion of this op." % str(s_new_out.owner.op))
return False return False
#create the composite op. #create the composite op.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论