提交 0da09905 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix test.

上级 0294ed08
...@@ -542,7 +542,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r ...@@ -542,7 +542,7 @@ def basic_multinomialtest(f, steps, sample_size, target_pvals, prefix="", mean_r
avg_pvals += ival avg_pvals += ival
avg_pvals/= steps avg_pvals/= steps
print 'random?[:10]\n', f()[:10] print 'random?[:10]\n', numpy.asarray(f()[:10])
print prefix, 'mean', avg_pvals print prefix, 'mean', avg_pvals
print numpy.mean(abs(avg_pvals - target_pvals))# < mean_rtol, 'bad mean? %s %s' % (str(avg_pvals), str(target_pvals)) print numpy.mean(abs(avg_pvals - target_pvals))# < mean_rtol, 'bad mean? %s %s' % (str(avg_pvals), str(target_pvals))
print prefix, 'time', dt print prefix, 'time', dt
...@@ -584,9 +584,10 @@ def test_multinomial(): ...@@ -584,9 +584,10 @@ def test_multinomial():
# We give the number of streams to avoid a warning. # We give the number of streams to avoid a warning.
n = R.multinomial(pvals=pvals, dtype='float32', nstreams=30 * 256) n = R.multinomial(pvals=pvals, dtype='float32', nstreams=30 * 256)
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.sandbox.cuda.basic_ops.gpu_from_host(n), [],
borrow=True), mode=mode_.including('gpu')) theano.sandbox.cuda.basic_ops.gpu_from_host(n),
mode=mode_.including('gpu'))
theano.printing.debugprint(f) theano.printing.debugprint(f)
gpu_out = f() gpu_out = f()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论