提交 567ceab6 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #2341 from abergeron/fix_buildbot

Fix buildbot error from #2329
...@@ -1093,10 +1093,9 @@ def test_inc_subtensor(): ...@@ -1093,10 +1093,9 @@ def test_inc_subtensor():
yval = numpy.asarray([[10, 10, 10], [10, 10, 10], [10, 10, 10]], yval = numpy.asarray([[10, 10, 10], [10, 10, 10], [10, 10, 10]],
dtype='float32') dtype='float32')
expr = T.inc_subtensor(x[:, 1:3], y[:, 1:3]) expr = T.inc_subtensor(x[:, 1:3], y[:, 1:3])
f = theano.function([x, y], expr, mode=mode_with_gpu) f = theano.function([x, y], expr, mode=mode_with_gpu)
assert sum([isinstance(node.op, cuda.GpuSubtensor)
for node in f.maker.fgraph.toposort()]) == 1
assert sum([isinstance(node.op, cuda.GpuIncSubtensor) and assert sum([isinstance(node.op, cuda.GpuIncSubtensor) and
node.op.set_instead_of_inc==False node.op.set_instead_of_inc==False
for node in f.maker.fgraph.toposort()]) == 1 for node in f.maker.fgraph.toposort()]) == 1
...@@ -1114,8 +1113,6 @@ def test_set_subtensor(): ...@@ -1114,8 +1113,6 @@ def test_set_subtensor():
dtype='float32') dtype='float32')
expr = T.set_subtensor(x[:, 1:3], y[:, 1:3]) expr = T.set_subtensor(x[:, 1:3], y[:, 1:3])
f = theano.function([x, y], expr, mode=mode_with_gpu) f = theano.function([x, y], expr, mode=mode_with_gpu)
assert sum([isinstance(node.op, cuda.GpuSubtensor)
for node in f.maker.fgraph.toposort()]) == 1
assert sum([isinstance(node.op, cuda.GpuIncSubtensor) and assert sum([isinstance(node.op, cuda.GpuIncSubtensor) and
node.op.set_instead_of_inc == True node.op.set_instead_of_inc == True
for node in f.maker.fgraph.toposort()]) == 1 for node in f.maker.fgraph.toposort()]) == 1
......
...@@ -434,7 +434,7 @@ class T_Scan(unittest.TestCase): ...@@ -434,7 +434,7 @@ class T_Scan(unittest.TestCase):
output, output,
updates=updates, updates=updates,
allow_input_downcast=True, allow_input_downcast=True,
mode=mode) mode=mode_with_gpu)
# get random initial values # get random initial values
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论