提交 1c7f6acc authored 作者: Frederic's avatar Frederic

Make optimization faster in debugmode.

We already disable the stack_trace_on_call in normal compilation.
上级 ca112010
...@@ -2116,14 +2116,17 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions ...@@ -2116,14 +2116,17 @@ class _Maker(FunctionMaker): # inheritance buys a few helper functions
# optimize the fgraph # optimize the fgraph
compute_test_value_orig = theano.config.compute_test_value compute_test_value_orig = theano.config.compute_test_value
add_stack_trace_on_call = gof.Op.add_stack_trace_on_call
try: try:
theano.config.compute_test_value = theano.config.compute_test_value_opt theano.config.compute_test_value = theano.config.compute_test_value_opt
gof.Op.add_stack_trace_on_call = False # Should it be 0 == i?
optimizer(fgraph) optimizer(fgraph)
theano.compile.function_module.insert_deepcopy(fgraph, inputs, theano.compile.function_module.insert_deepcopy(fgraph, inputs,
outputs + additional_outputs) outputs + additional_outputs)
finally: finally:
theano.config.compute_test_value = compute_test_value_orig theano.config.compute_test_value = compute_test_value_orig
gof.Op.add_stack_trace_on_call = add_stack_trace_on_call
if i: if i:
li = fgraph.equivalence_tracker.event_list li = fgraph.equivalence_tracker.event_list
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论