提交 d7fa2289 authored 作者: Frederic's avatar Frederic

Do not use inplace tag for gpu opt

上级 08ec1f6f
...@@ -436,13 +436,13 @@ def use(device, ...@@ -436,13 +436,13 @@ def use(device,
cuda_enabled = True cuda_enabled = True
if default_to_move_computation_to_gpu: if default_to_move_computation_to_gpu:
# Do not add inplace tag here. We do not want to
# enable/disable gpu opt based on the inplace tag.
optdb.add_tags('gpu_opt', optdb.add_tags('gpu_opt',
'fast_compile', 'fast_compile',
'fast_run', 'fast_run')
'inplace')
optdb.add_tags('gpu_after_fusion', optdb.add_tags('gpu_after_fusion',
'fast_run', 'fast_run')
'inplace')
if force: if force:
try: try:
......
...@@ -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', 'inplace') optdb.add_tags('gpuarray_opt', '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)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论