提交 9f04330b authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add option to bypass error for GPU

上级 ae01ef54
...@@ -251,8 +251,9 @@ class ProfileStats(object): ...@@ -251,8 +251,9 @@ class ProfileStats(object):
# param is called flag_time_thunks because most other attributes with time # param is called flag_time_thunks because most other attributes with time
# in the name are times *of* something, rather than configuration flags. # in the name are times *of* something, rather than configuration flags.
def __init__(self, atexit_print=True, flag_time_thunks=None, **kwargs): def __init__(self, atexit_print=True, flag_time_thunks=None,
if (config.profile and gpu_checks=True, **kwargs):
if (config.profile and gpu_checks and
((hasattr(theano, 'sandbox') and ((hasattr(theano, 'sandbox') and
hasattr(theano.sandbox, 'cuda') and hasattr(theano.sandbox, 'cuda') and
theano.sandbox.cuda.cuda_enabled) or ( theano.sandbox.cuda.cuda_enabled) or (
...@@ -266,7 +267,7 @@ class ProfileStats(object): ...@@ -266,7 +267,7 @@ class ProfileStats(object):
" You must set the environment variable" " You must set the environment variable"
" CUDA_LAUNCH_BLOCKING to 1 to tell the CUDA driver to" " CUDA_LAUNCH_BLOCKING to 1 to tell the CUDA driver to"
" synchronize the execution to get a meaningful profile.") " synchronize the execution to get a meaningful profile.")
if (config.profile and if (config.profile and gpu_checks and
hasattr(theano, 'gpuarray') and hasattr(theano, 'gpuarray') and
theano.gpuarray.pygpu_activated and theano.gpuarray.pygpu_activated and
not config.profiling.ignore_first_call): not config.profiling.ignore_first_call):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论