提交 59f671e2 authored 作者: Simon Lefrancois's avatar Simon Lefrancois 提交者: GitHub

Merge pull request #5363 from nouiz/fix_test

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