提交 919f6779 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #2886 from carriepl/fix_buildbot

Fix scan test for float64
...@@ -1374,7 +1374,8 @@ class T_Scan(unittest.TestCase): ...@@ -1374,7 +1374,8 @@ class T_Scan(unittest.TestCase):
raise SkipTest('Optional package cuda disabled') raise SkipTest('Optional package cuda disabled')
rs = theano.sandbox.rng_mrg.MRG_RandomStreams(use_cuda=True) 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,), dtype="float32"),
n_steps=3)
cPickle.loads(cPickle.dumps(output)) cPickle.loads(cPickle.dumps(output))
# Also ensure that, after compilation, the Scan has been moved # Also ensure that, after compilation, the Scan has been moved
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论