提交 5f0a1536 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix buildbot.

上级 8b5986f9
...@@ -8,6 +8,10 @@ from theano.sandbox.rng_mrg import MRG_RandomStreams ...@@ -8,6 +8,10 @@ from theano.sandbox.rng_mrg import MRG_RandomStreams
#def test_rng_mrg_cpu(): #def test_rng_mrg_cpu():
#TODO: make tests work when no flags gived. Now need: THEANO_FLAGS=device=gpu0,floatX=float32 #TODO: make tests work when no flags gived. Now need: THEANO_FLAGS=device=gpu0,floatX=float32
#TODO: bug fix test_normal0, in normal() fct, n_samples currently need to be numpy.prod(size) not self.n_streams(size) #TODO: bug fix test_normal0, in normal() fct, n_samples currently need to be numpy.prod(size) not self.n_streams(size)
mode = theano.config.mode
#THIS IS THEIR AS THEIR IS A MEMORY LINK in perform THAT WILL MAKE THE BUILDBOT DON'T WORK
mode='FAST_RUN'#TODO: remove this
def test_rng0(): def test_rng0():
def basictest(f, steps, prefix=""): def basictest(f, steps, prefix=""):
...@@ -48,10 +52,10 @@ def test_rng0(): ...@@ -48,10 +52,10 @@ def test_rng0():
R = MRG_RandomStreams(234, use_cuda=False) R = MRG_RandomStreams(234, use_cuda=False)
u = R.uniform(size=sample_size) u = R.uniform(size=sample_size)
f = theano.function([], u) f = theano.function([], u, mode=mode)
theano.printing.debugprint(f) theano.printing.debugprint(f)
print 'random?[:10]\n', f()[0,0:10] print 'random?[:10]\n', f()[0,0:10]
basictest(f, 1000, prefix='mrg ') basictest(f, steps, prefix='mrg cpu')
print '' print ''
print 'ON GPU:' print 'ON GPU:'
...@@ -60,17 +64,17 @@ def test_rng0(): ...@@ -60,17 +64,17 @@ def test_rng0():
assert u.dtype == 'float32' #well, it's really that this test w GPU doesn't make sense otw assert u.dtype == 'float32' #well, it's really that this test w GPU doesn't make sense otw
f = theano.function([], theano.Out( f = theano.function([], theano.Out(
theano.sandbox.cuda.basic_ops.gpu_from_host(u), theano.sandbox.cuda.basic_ops.gpu_from_host(u),
borrow=True)) borrow=True), mode=mode)
theano.printing.debugprint(f) theano.printing.debugprint(f)
print 'random?[:10]\n', numpy.asarray(f())[0,0:10] print 'random?[:10]\n', numpy.asarray(f())[0,0:10]
basictest(f, 1000, prefix='mrg ') basictest(f, steps, prefix='mrg gpu')
print '' print ''
print 'ON CPU w NUMPY:' print 'ON CPU w NUMPY:'
RR = theano.tensor.shared_randomstreams.RandomStreams(234) RR = theano.tensor.shared_randomstreams.RandomStreams(234)
uu = RR.uniform(size=sample_size) uu = RR.uniform(size=sample_size)
ff = theano.function([], uu) ff = theano.function([], uu, mode=mode)
basictest(ff, 1000, prefix='numpy') basictest(ff, 1000, prefix='numpy')
...@@ -109,7 +113,7 @@ def test_normal0(): ...@@ -109,7 +113,7 @@ def test_normal0():
R = MRG_RandomStreams(234, use_cuda=False) R = MRG_RandomStreams(234, use_cuda=False)
n = R.normal(size=sample_size, avg=-5.0, std=2.0) n = R.normal(size=sample_size, avg=-5.0, std=2.0)
f = theano.function([], n) f = theano.function([], n, mode=mode)
theano.printing.debugprint(f) theano.printing.debugprint(f)
print 'random?[:10]\n', f()[0,0:10] print 'random?[:10]\n', f()[0,0:10]
basictest(f, 50, -5.0, 2.0, prefix='mrg ') basictest(f, 50, -5.0, 2.0, prefix='mrg ')
...@@ -127,7 +131,7 @@ def test_normal0(): ...@@ -127,7 +131,7 @@ def test_normal0():
assert n.dtype == 'float32' #well, it's really that this test w GPU doesn't make sense otw assert n.dtype == 'float32' #well, it's really that this test w GPU doesn't make sense otw
f = theano.function([], theano.Out( f = theano.function([], theano.Out(
theano.sandbox.cuda.basic_ops.gpu_from_host(n), theano.sandbox.cuda.basic_ops.gpu_from_host(n),
borrow=True)) borrow=True), mode=mode)
theano.printing.debugprint(f) theano.printing.debugprint(f)
sys.stdout.flush() sys.stdout.flush()
...@@ -142,7 +146,7 @@ def test_normal0(): ...@@ -142,7 +146,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) ff = theano.function([], nn, mode=mode)
basictest(ff, 50, -5.0, 2.0, prefix='numpy ') basictest(ff, 50, -5.0, 2.0, prefix='numpy ')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论