提交 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):
@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):
return [gpugemm_no_inplace(*inputs)]
@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):
return [gpugemm_no_inplace(*inputs)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论