提交 0c6020f2 authored 作者: Frederic Bastien's avatar Frederic Bastien

backport to python 2.4

上级 c9267121
...@@ -777,7 +777,10 @@ class T_using_gpu(unittest.TestCase): ...@@ -777,7 +777,10 @@ class T_using_gpu(unittest.TestCase):
r = f() r = f()
print 'Looping %d times took'%iters, time.time() - t0, 'seconds' print 'Looping %d times took'%iters, time.time() - t0, 'seconds'
print 'Result is', r print 'Result is', r
print 'Used the','cpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) else 'gpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]):
print 'Used the cpu'
else:
print 'Used the gpu'
if theano.config.device.find('gpu') > -1: if theano.config.device.find('gpu') > -1:
assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()])
else: else:
...@@ -806,7 +809,10 @@ class T_using_gpu(unittest.TestCase): ...@@ -806,7 +809,10 @@ class T_using_gpu(unittest.TestCase):
print 'Looping %d times took'%iters, time.time() - t0, 'seconds' print 'Looping %d times took'%iters, time.time() - t0, 'seconds'
print 'Result is', r print 'Result is', r
print 'Numpy result is', numpy.asarray(r) print 'Numpy result is', numpy.asarray(r)
print 'Used the','cpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) else 'gpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]):
print 'Used the cpu'
else:
print 'Used the gpu'
assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()])
...@@ -837,7 +843,10 @@ class T_using_gpu(unittest.TestCase): ...@@ -837,7 +843,10 @@ class T_using_gpu(unittest.TestCase):
print 'Looping %d times took'%iters, time.time() - t0, 'seconds' print 'Looping %d times took'%iters, time.time() - t0, 'seconds'
print 'Result is', r print 'Result is', r
print 'Numpy result is', numpy.asarray(r) print 'Numpy result is', numpy.asarray(r)
print 'Used the','cpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) else 'gpu' if numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]):
print 'Used the cpu'
else:
print 'Used the gpu'
assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()]) assert not numpy.any( [isinstance(x.op,T.Elemwise) for x in f.maker.env.toposort()])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论