提交 bf4cd600 authored 作者: sebastien-j's avatar sebastien-j

Use 'float32' to force execution on GPU

上级 dbe722ba
......@@ -318,7 +318,7 @@ def test_GPU_nstreams_limit():
R = MRG_RandomStreams(seed=seed, use_cuda=True)
def eval_uniform(size, nstreams):
return R.uniform(size=size, nstreams=nstreams).eval()
return R.uniform(size=size, nstreams=nstreams, dtype='float32').eval()
eval_uniform((10,), 2**20)
assert_raises(ValueError, eval_uniform, (10,), 2**20 + 1)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论