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

Move the definition of gpuarray.sync

上级 31e6600f
...@@ -177,6 +177,13 @@ AddConfigVar('nvcc.fastmath', ...@@ -177,6 +177,13 @@ AddConfigVar('nvcc.fastmath',
# if theano.sandbox.cuda is loaded or not. # if theano.sandbox.cuda is loaded or not.
in_c_key=False) in_c_key=False)
AddConfigVar('gpuarray.sync',
"""If True, every op will make sure its work is done before
returning. Setting this to True will slow down execution,
but give much more accurate results in profiling.""",
BoolParam(False),
in_c_key=True)
AddConfigVar('dnn.conv.workmem', AddConfigVar('dnn.conv.workmem',
"This flag is deprecated; use dnn.conv.algo_fwd.", "This flag is deprecated; use dnn.conv.algo_fwd.",
EnumStr(''), EnumStr(''),
......
...@@ -19,13 +19,6 @@ try: ...@@ -19,13 +19,6 @@ try:
except ImportError: except ImportError:
pygpu = None pygpu = None
AddConfigVar('gpuarray.sync',
"""If True, every op will make sure its work is done before
returning. Setting this to True will slow down execution,
but give much more accurate results in profiling.""",
BoolParam(False),
in_c_key=True)
# This is for documentation not to depend on the availability of pygpu # This is for documentation not to depend on the availability of pygpu
from .type import (GpuArrayType, GpuArrayVariable, GpuArrayConstant, from .type import (GpuArrayType, GpuArrayVariable, GpuArrayConstant,
GpuArraySharedVariable, gpuarray_shared_constructor) GpuArraySharedVariable, gpuarray_shared_constructor)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论