提交 ecba472d authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Normalize the optimization tag to use.

上级 0a162810
...@@ -56,7 +56,7 @@ if pygpu: ...@@ -56,7 +56,7 @@ if pygpu:
init_dev(config.device) init_dev(config.device)
import theano.compile import theano.compile
theano.compile.shared_constructor(gpuarray_shared_constructor) theano.compile.shared_constructor(gpuarray_shared_constructor)
optdb.add_tags('gpuarray_opt', 'fast_run', 'fast_compile') optdb.add_tags('gpuarray', 'fast_run', 'fast_compile')
elif config.gpuarray.init_device != '': elif config.gpuarray.init_device != '':
init_dev(config.gpuarray.init_device) init_dev(config.gpuarray.init_device)
......
...@@ -1588,7 +1588,7 @@ optdb.register('local_dnna_conv_inplace', ...@@ -1588,7 +1588,7 @@ optdb.register('local_dnna_conv_inplace',
local_dnn_convgw_inplace, local_dnn_convgw_inplace,
local_dnn_convgi_inplace, local_dnn_convgi_inplace,
name="local_dnn_conv_inplace"), name="local_dnn_conv_inplace"),
70.0, 'fast_run', 'inplace', 'gpu', 'cudnn') 70.0, 'fast_run', 'inplace', 'gpuarray', 'cudnn')
@register_opt('cudnn') @register_opt('cudnn')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论