提交 70083b27 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix tests ni FAST_COMPILE mode.

上级 609593b4
...@@ -264,7 +264,7 @@ def test_consistency_GPU_parallel(): ...@@ -264,7 +264,7 @@ def test_consistency_GPU_parallel():
# We need the sample back in the main memory # We need the sample back in the main memory
cpu_sample = tensor.as_tensor_variable(sample) cpu_sample = tensor.as_tensor_variable(sample)
f = theano.function([], cpu_sample) f = theano.function([], cpu_sample, mode=mode)
for k in range(n_samples): for k in range(n_samples):
s = f() s = f()
...@@ -351,6 +351,10 @@ def test_rng0(): ...@@ -351,6 +351,10 @@ def test_rng0():
def test_normal0(): def test_normal0():
if config.mode == 'FAST_COMPILE':
mode = 'FAST_RUN'
else:
mode = config.mode
def basictest(f, steps, target_avg, target_std, prefix=""): def basictest(f, steps, target_avg, target_std, prefix=""):
dt = 0.0 dt = 0.0
avg_std = 0.0 avg_std = 0.0
...@@ -414,7 +418,7 @@ def test_normal0(): ...@@ -414,7 +418,7 @@ def test_normal0():
RR = theano.tensor.shared_randomstreams.RandomStreams(234) RR = theano.tensor.shared_randomstreams.RandomStreams(234)
nn = RR.normal(size=sample_size, avg=-5.0, std=2.0) nn = RR.normal(size=sample_size, avg=-5.0, std=2.0)
ff = theano.function([], nn, mode=mode) ff = theano.function([], nn)
basictest(ff, 50, -5.0, 2.0, prefix='numpy ') basictest(ff, 50, -5.0, 2.0, prefix='numpy ')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论