提交 5a7ae750 authored 作者: Frederic Bastien's avatar Frederic Bastien

Raise in the new back-end as in the old about CUDA_LAUNCH_BLOCKING

上级 bfde042c
...@@ -206,9 +206,12 @@ class ProfileStats(object): ...@@ -206,9 +206,12 @@ 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, **kwargs):
if (hasattr(theano, 'sandbox') and if (config.profile and
((hasattr(theano, 'sandbox') and
hasattr(theano.sandbox, 'cuda') and hasattr(theano.sandbox, 'cuda') and
theano.sandbox.cuda.cuda_enabled): theano.sandbox.cuda.cuda_enabled)) or(
hasattr(theano, 'gpuarray') and
theano.gpuarray.pygpu_activated)):
if os.environ.get('CUDA_LAUNCH_BLOCKING', '0') != '1': if os.environ.get('CUDA_LAUNCH_BLOCKING', '0') != '1':
raise Exception( raise Exception(
"You are running the Theano profiler with CUDA enabled." "You are running the Theano profiler with CUDA enabled."
......
...@@ -752,9 +752,11 @@ class VM_Linker(link.LocalLinker): ...@@ -752,9 +752,11 @@ class VM_Linker(link.LocalLinker):
""" """
if (config.profile and if (config.profile and
hasattr(theano, 'sandbox') and ((hasattr(theano, 'sandbox') and
hasattr(theano.sandbox, 'cuda') and hasattr(theano.sandbox, 'cuda') and
theano.sandbox.cuda.cuda_enabled): theano.sandbox.cuda.cuda_enabled)) or(
hasattr(theano, 'gpuarray') and
theano.gpuarray.pygpu_activated)):
if os.environ.get('CUDA_LAUNCH_BLOCKING', '0') != '1': if os.environ.get('CUDA_LAUNCH_BLOCKING', '0') != '1':
raise Exception( raise Exception(
"You are running the Theano profiler with CUDA enabled." "You are running the Theano profiler with CUDA enabled."
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论