提交 06d5557f authored 作者: James Bergstra's avatar James Bergstra

more precise linker specification in test_scan.test_speed_rnn

上级 cca9e5d6
...@@ -2239,7 +2239,7 @@ def test_speed_rnn(): ...@@ -2239,7 +2239,7 @@ def test_speed_rnn():
sequences=[s_r[1:]], sequences=[s_r[1:]],
outputs_info=tensor.constant(r[0])) outputs_info=tensor.constant(r[0]))
assert not updates assert not updates
f = theano.function([s_r], s_y, mode='FAST_RUN_NOGC') f = theano.function([s_r], s_y, mode=theano.Mode(linker='c|py_nogc'))
t2 = time.time() t2 = time.time()
f(r) f(r)
...@@ -2266,8 +2266,8 @@ def test_speed_rnn(): ...@@ -2266,8 +2266,8 @@ def test_speed_rnn():
mode=theano.Mode(linker='cvm')) mode=theano.Mode(linker='cvm'))
theano.printing.debugprint(f ) theano.printing.debugprint(f )
f_fn = f.fn f_fn = f.fn
t2 = time.time()
print f_fn print f_fn
t2 = time.time()
f_fn(n_calls=L-2) f_fn(n_calls=L-2)
f() #999 to update the profiling timers f() #999 to update the profiling timers
t3 = time.time() t3 = time.time()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论