提交 393940fd authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix missed calls to alpha/output_merge

上级 25ca446b
...@@ -645,13 +645,13 @@ def local_gpua_hgemm(node): ...@@ -645,13 +645,13 @@ def local_gpua_hgemm(node):
@register_opt() @register_opt()
@alpha_merge(GpuGemm, alpha_in=1, beta_in=4, nd=2) @alpha_merge(GpuGemm, alpha_in=1, beta_in=4)
def local_gpuagemm_alpha_merge(node, *inputs): def local_gpuagemm_alpha_merge(node, *inputs):
return [gpugemm_no_inplace(*inputs)] return [gpugemm_no_inplace(*inputs)]
@register_opt() @register_opt()
@output_merge(GpuGemm, alpha_in=1, beta_in=4, out_in=0, nd=2) @output_merge(GpuGemm, alpha_in=1, beta_in=4, out_in=0)
def local_gpuagemm_output_merge(node, *inputs): def local_gpuagemm_output_merge(node, *inputs):
return [gpugemm_no_inplace(*inputs)] return [gpugemm_no_inplace(*inputs)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论