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

Do not use inplace tag for gpu opt

上级 08ec1f6f
......@@ -436,13 +436,13 @@ def use(device,
cuda_enabled = True
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',
'fast_compile',
'fast_run',
'inplace')
'fast_run')
optdb.add_tags('gpu_after_fusion',
'fast_run',
'inplace')
'fast_run')
if force:
try:
......
......@@ -56,7 +56,7 @@ if pygpu:
init_dev(config.device)
import theano.compile
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 != '':
init_dev(config.gpuarray.init_device)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论