提交 daaba020 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix test that fails on the buildbot.

上级 a62b8883
...@@ -823,7 +823,8 @@ def test_blocksparse_gpu_gemv_opt(): ...@@ -823,7 +823,8 @@ def test_blocksparse_gpu_gemv_opt():
f = theano.function([W, h, iIdx, b, oIdx], o, mode=mode_with_gpu) f = theano.function([W, h, iIdx, b, oIdx], o, mode=mode_with_gpu)
assert isinstance(f.maker.fgraph.toposort()[-2].op, GpuSparseBlockGemv) assert sum(1 for n in f.maker.fgraph.apply_nodes
if isinstance(n.op, GpuSparseBlockGemv)) == 1
def test_blocksparse_gpu_outer_opt(): def test_blocksparse_gpu_outer_opt():
...@@ -839,7 +840,8 @@ def test_blocksparse_gpu_outer_opt(): ...@@ -839,7 +840,8 @@ def test_blocksparse_gpu_outer_opt():
wrt=W)], wrt=W)],
mode=mode_with_gpu) mode=mode_with_gpu)
assert isinstance(f.maker.fgraph.toposort()[-2].op, GpuSparseBlockOuter) assert sum(1 for n in f.maker.fgraph.apply_nodes
if isinstance(n.op, GpuSparseBlockOuter)) == 1
class test_diag(theano.tensor.tests.test_nlinalg.test_diag): class test_diag(theano.tensor.tests.test_nlinalg.test_diag):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论