提交 78158149 authored 作者: Frederic's avatar Frederic

flake8

上级 b4240f3f
...@@ -512,14 +512,14 @@ def test_local_gpu_elemwise_careduce(): ...@@ -512,14 +512,14 @@ def test_local_gpu_elemwise_careduce():
assert len(topo) == 3 assert len(topo) == 3
assert topo[1].op.pre_scalar_op == theano.scalar.sqr assert topo[1].op.pre_scalar_op == theano.scalar.sqr
data = numpy.random.rand(3, 4).astype('float32') data = numpy.random.rand(3, 4).astype('float32')
utt.assert_allclose(f(data), (data*data).sum()) utt.assert_allclose(f(data), (data * data).sum())
o = (x * x).sum(axis=1) o = (x * x).sum(axis=1)
f = theano.function([x], o, mode=mode_with_gpu) f = theano.function([x], o, mode=mode_with_gpu)
topo = f.maker.fgraph.toposort() topo = f.maker.fgraph.toposort()
assert len(topo) == 3 assert len(topo) == 3
assert topo[1].op.pre_scalar_op == theano.scalar.sqr assert topo[1].op.pre_scalar_op == theano.scalar.sqr
utt.assert_allclose(f(data), (data*data).sum(axis=1)) utt.assert_allclose(f(data), (data * data).sum(axis=1))
def test_huge_elemwise_fusion(): def test_huge_elemwise_fusion():
......
...@@ -213,14 +213,14 @@ def test_local_gpu_elemwise_careduce(): ...@@ -213,14 +213,14 @@ def test_local_gpu_elemwise_careduce():
assert len(topo) == 3 assert len(topo) == 3
assert topo[1].op.pre_scalar_op == theano.scalar.sqr assert topo[1].op.pre_scalar_op == theano.scalar.sqr
data = numpy.random.rand(3, 4).astype(theano.config.floatX) data = numpy.random.rand(3, 4).astype(theano.config.floatX)
utt.assert_allclose(f(data), (data*data).sum()) utt.assert_allclose(f(data), (data * data).sum())
o = (x * x).sum(axis=1) o = (x * x).sum(axis=1)
f = theano.function([x], o, mode=mode_with_gpu) f = theano.function([x], o, mode=mode_with_gpu)
topo = f.maker.fgraph.toposort() topo = f.maker.fgraph.toposort()
assert len(topo) == 3 assert len(topo) == 3
assert topo[1].op.pre_scalar_op == theano.scalar.sqr assert topo[1].op.pre_scalar_op == theano.scalar.sqr
utt.assert_allclose(f(data), (data*data).sum(axis=1)) utt.assert_allclose(f(data), (data * data).sum(axis=1))
def test_local_gpu_subtensor(): def test_local_gpu_subtensor():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论