提交 c5b8fcf6 authored 作者: Frederic's avatar Frederic

Small opt speed up

上级 c3f6a4c5
...@@ -345,7 +345,7 @@ def use_c_ger(node): ...@@ -345,7 +345,7 @@ def use_c_ger(node):
@local_optimizer([CGer(False)]) @local_optimizer([CGer(False)])
def make_c_ger_destructive(node): def make_c_ger_destructive(node):
if node.op == cger_no_inplace: if isinstance(node.op, CGer) and not node.op.destructive:
return [cger_inplace(*node.inputs)] return [cger_inplace(*node.inputs)]
...@@ -800,7 +800,7 @@ def use_c_gemv(node): ...@@ -800,7 +800,7 @@ def use_c_gemv(node):
@local_optimizer([CGemv(inplace=False)]) @local_optimizer([CGemv(inplace=False)])
def make_c_gemv_destructive(node): def make_c_gemv_destructive(node):
if node.op == cgemv_no_inplace: if isinstance(node.op, CGemv) and not node.op.inplace:
return [cgemv_inplace(*node.inputs)] return [cgemv_inplace(*node.inputs)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论