提交 684eb636 authored 作者: Frederic Bastien's avatar Frederic Bastien

This test that we don't move 4d tensor to the GPU. Problem introduced in gh-5317.

上级 9bff4ddc
...@@ -260,4 +260,4 @@ class TestGpuCumOp(theano.tensor.tests.test_extra_ops.TestCumOp): ...@@ -260,4 +260,4 @@ class TestGpuCumOp(theano.tensor.tests.test_extra_ops.TestCumOp):
x = T.ftensor4('x') x = T.ftensor4('x')
f = theano.function([x], op_class(axis=1)(x), mode=self.mode) f = theano.function([x], op_class(axis=1)(x), mode=self.mode)
assert [n for n in f.maker.fgraph.toposort() assert [n for n in f.maker.fgraph.toposort()
if isinstance(n.op, GpuCumOp)] if isinstance(n.op, CumOp)]
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论