提交 5e2e87e0 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix a change to gpuarray conv from gh-5069. dnn conv wasn't working anymore in fast_compile.

上级 173826a7
...@@ -1784,7 +1784,7 @@ optdb.register('gpua_scanOp_make_inplace', ...@@ -1784,7 +1784,7 @@ optdb.register('gpua_scanOp_make_inplace',
# which ones take precedence over others. # which ones take precedence over others.
abstractconv_groupopt = theano.gof.optdb.LocalGroupDB() abstractconv_groupopt = theano.gof.optdb.LocalGroupDB()
abstractconv_groupopt.__name__ = "gpuarray_abstractconv_opts" abstractconv_groupopt.__name__ = "gpuarray_abstractconv_opts"
register_opt()(abstractconv_groupopt) register_opt('fast_compile')(abstractconv_groupopt)
# cuDNN is first, but only registered if cuDNN is available. # cuDNN is first, but only registered if cuDNN is available.
# (we import these opts here instead of at the top of this file # (we import these opts here instead of at the top of this file
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论