提交 a29c31d3 authored 作者: Pascal Lamblin's avatar Pascal Lamblin 提交者: GitHub

Merge pull request #5108 from nouiz/fast_compile_fix

Fix a change to gpuarray conv from gh-5069. dnn conv wasn't working a…
...@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论