提交 e33c0f75 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Apply the optimization on cgemv, not gemv

上级 30f824ad
......@@ -600,8 +600,8 @@ def use_c_gemv(node):
@local_optimizer([CGemv(inplace=False)])
def make_c_gemv_destructive(node):
if node.op == gemv_no_inplace:
return [gemv_inplace(*node.inputs)]
if node.op == cgemv_no_inplace:
return [cgemv_inplace(*node.inputs)]
####### ####### #######
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论