提交 2bd5f25d authored 作者: --global's avatar --global

Force test to use GPU

上级 42372d05
...@@ -1372,7 +1372,7 @@ class T_Scan(unittest.TestCase): ...@@ -1372,7 +1372,7 @@ class T_Scan(unittest.TestCase):
if not cuda.cuda_available: if not cuda.cuda_available:
raise SkipTest('Optional package cuda disabled') raise SkipTest('Optional package cuda disabled')
rs = theano.sandbox.rng_mrg.MRG_RandomStreams() rs = theano.sandbox.rng_mrg.MRG_RandomStreams(use_cuda=True)
output, _ = theano.scan(lambda : rs.uniform((3,)), n_steps=3) output, _ = theano.scan(lambda : rs.uniform((3,)), n_steps=3)
cPickle.loads(cPickle.dumps(output)) cPickle.loads(cPickle.dumps(output))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论