提交 6a3cb283 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix having gpuarray as the default compute backend not working due to removed optimization tags.

上级 1e91880d
...@@ -35,7 +35,6 @@ if pygpu: ...@@ -35,7 +35,6 @@ if pygpu:
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', 'inplace') optdb.add_tags('gpuarray_opt', 'fast_run', 'inplace')
optdb.add_tags('gpuarray_after_fusion', 'fast_run', 'inplace')
elif config.gpuarray.init_device != '': elif config.gpuarray.init_device != '':
init_dev(config.gpuarray.init_device) init_dev(config.gpuarray.init_device)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论