提交 2c6b9b06 authored 作者: Frederic Bastien's avatar Frederic Bastien

Disable by default the memory profiling in ProfileMode and fix the use of that flags.

上级 2d7879b5
...@@ -25,7 +25,7 @@ AddConfigVar('ProfileMode.min_memory_size', ...@@ -25,7 +25,7 @@ AddConfigVar('ProfileMode.min_memory_size',
AddConfigVar('ProfileMode.profile_memory', AddConfigVar('ProfileMode.profile_memory',
"""Enable profiling of memory used by Theano functions""", """Enable profiling of memory used by Theano functions""",
BoolParam(True)) BoolParam(False))
class Profile_Maker(FunctionMaker): class Profile_Maker(FunctionMaker):
def create(self, input_storage=None, trustme=False): def create(self, input_storage=None, trustme=False):
...@@ -146,7 +146,7 @@ class ProfileMode(Mode): ...@@ -146,7 +146,7 @@ class ProfileMode(Mode):
if isinstance(linker, str) or linker is None: if isinstance(linker, str) or linker is None:
linker = predefined_linkers[linker] linker = predefined_linkers[linker]
if config.ProfileMode.profile_memory: if not config.ProfileMode.profile_memory:
p_thunk = profile_thunk p_thunk = profile_thunk
else: else:
p_thunk = profile_thunk2 p_thunk = profile_thunk2
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论