提交 7dcb1922 authored 作者: Frederic Bastien's avatar Frederic Bastien

Move conv to the GPU in fast_compile in the new back-end

上级 041a384e
......@@ -946,7 +946,7 @@ def local_lift_abstractconv2d(node, context_name):
return [node.op(*inps)]
# Register this here so that it goes after the abstract lifting
register_opt()(conv_groupopt)
register_opt('fast_compile')(conv_groupopt)
@register_opt("low_memory")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论