提交 ce65dc03 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Small changes in info message in rng mrg test

上级 20a0882d
...@@ -349,10 +349,9 @@ def test_uniform(): ...@@ -349,10 +349,9 @@ def test_uniform():
theano.printing.debugprint(f) theano.printing.debugprint(f)
cpu_out = f(*input) cpu_out = f(*input)
print 'random?[:10]\n' print 'CPU: random?[:10], random?[-10:]'
print cpu_out[0,0:10] print cpu_out[0,0:10]
print cpu_out[-1,-10:] print cpu_out[-1,-10:]
#print 'random?[-1,-10:]\n', cpu_out[-1,-10:]
basictest(f, steps, sample_size, prefix='mrg cpu', inputs=input) basictest(f, steps, sample_size, prefix='mrg cpu', inputs=input)
if mode!='FAST_COMPILE' and cuda_available: if mode!='FAST_COMPILE' and cuda_available:
...@@ -369,10 +368,9 @@ def test_uniform(): ...@@ -369,10 +368,9 @@ def test_uniform():
theano.printing.debugprint(f) theano.printing.debugprint(f)
gpu_out = numpy.asarray(f(*input)) gpu_out = numpy.asarray(f(*input))
print 'random?[:10]\n' print 'GPU: random?[:10], random?[-10:]'
print gpu_out[0,0:10] print gpu_out[0,0:10]
print gpu_out[-1,-10:] print gpu_out[-1,-10:]
#print 'random?[-1,-10:]\n', gpu_out[-1,-10:]
basictest(f, steps, sample_size, prefix='mrg gpu', inputs=input) basictest(f, steps, sample_size, prefix='mrg gpu', inputs=input)
numpy.testing.assert_array_almost_equal(cpu_out, gpu_out, decimal=6) numpy.testing.assert_array_almost_equal(cpu_out, gpu_out, decimal=6)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论