提交 840352dd authored 作者: Frederic Bastien's avatar Frederic Bastien

Add Theano flag traceback.compile_limit

上级 69338f33
...@@ -959,3 +959,16 @@ import theano and print the config variable, as in: ...@@ -959,3 +959,16 @@ import theano and print the config variable, as in:
Bool value, default: False Bool value, default: False
If set to True, will preload the C module cache at import time If set to True, will preload the C module cache at import time
.. attribute:: config.traceback.limit
Int value, default: 8
The number of user stack level to keep for variables.
.. attribute:: config.traceback.compile_limit
Bool value, default: 0
The number of user stack level to keep for variables during Theano
compilation.
...@@ -1492,7 +1492,7 @@ class FunctionMaker(object): ...@@ -1492,7 +1492,7 @@ class FunctionMaker(object):
# optimize the fgraph # optimize the fgraph
theano.config.compute_test_value = \ theano.config.compute_test_value = \
theano.config.compute_test_value_opt theano.config.compute_test_value_opt
theano.config.traceback.limit = 0 theano.config.traceback.limit = theano.config.traceback.compile_limit
start_optimizer = time.time() start_optimizer = time.time()
# now optimize the graph # now optimize the graph
...@@ -1683,7 +1683,7 @@ class FunctionMaker(object): ...@@ -1683,7 +1683,7 @@ class FunctionMaker(object):
start_import_time = theano.gof.cmodule.import_time start_import_time = theano.gof.cmodule.import_time
limit_orig = theano.config.traceback.limit limit_orig = theano.config.traceback.limit
try: try:
theano.config.traceback.limit = 0 theano.config.traceback.limit = theano.config.traceback.compile_limit
_fn, _i, _o = self.linker.make_thunk( _fn, _i, _o = self.linker.make_thunk(
input_storage=input_storage_lists, storage_map=storage_map) input_storage=input_storage_lists, storage_map=storage_map)
finally: finally:
......
...@@ -572,6 +572,16 @@ AddConfigVar( ...@@ -572,6 +572,16 @@ AddConfigVar(
IntParam(8), IntParam(8),
in_c_key=False) in_c_key=False)
AddConfigVar(
'traceback.compile_limit',
"The number of stack to trace to keep during compilation. -1 mean all.",
IntParam(0),
in_c_key=False)
AddConfigVar('experimental.mrg',
"Another random number generator that work on the gpu",
BoolParam(False))
AddConfigVar('experimental.unpickle_gpu_on_cpu', AddConfigVar('experimental.unpickle_gpu_on_cpu',
"Allow unpickling of pickled CudaNdarrays as numpy.ndarrays." "Allow unpickling of pickled CudaNdarrays as numpy.ndarrays."
"This is useful, if you want to open a CudaNdarray without " "This is useful, if you want to open a CudaNdarray without "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论