提交 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:
Bool value, default: False
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):
# optimize the fgraph
theano.config.compute_test_value = \
theano.config.compute_test_value_opt
theano.config.traceback.limit = 0
theano.config.traceback.limit = theano.config.traceback.compile_limit
start_optimizer = time.time()
# now optimize the graph
......@@ -1683,7 +1683,7 @@ class FunctionMaker(object):
start_import_time = theano.gof.cmodule.import_time
limit_orig = theano.config.traceback.limit
try:
theano.config.traceback.limit = 0
theano.config.traceback.limit = theano.config.traceback.compile_limit
_fn, _i, _o = self.linker.make_thunk(
input_storage=input_storage_lists, storage_map=storage_map)
finally:
......
......@@ -572,6 +572,16 @@ AddConfigVar(
IntParam(8),
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',
"Allow unpickling of pickled CudaNdarrays as numpy.ndarrays."
"This is useful, if you want to open a CudaNdarray without "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论