提交 582210c7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Remove check on GPU in DebugMode

The corresponding CPU test does the same
上级 fbb066c4
...@@ -481,13 +481,12 @@ def test_hostfromgpu_shape_i(): ...@@ -481,13 +481,12 @@ def test_hostfromgpu_shape_i():
def test_Gpujoin_inplace(): def test_Gpujoin_inplace():
"""Test Gpujoin to work inplace. # Test Gpujoin to work inplace.
#
This function tests the case when several elements are passed to the # This function tests the case when several elements are passed to the
Gpujoin function but all except one of them are empty. In this case # Gpujoin function but all except one of them are empty. In this case
Gpujoin should work inplace and the output should be the view of the # Gpujoin should work inplace and the output should be the view of the
non-empty element. # non-empty element.
"""
s = T.lscalar() s = T.lscalar()
data = np.array([3, 4, 5], dtype=theano.config.floatX) data = np.array([3, 4, 5], dtype=theano.config.floatX)
x = gpuarray_shared_constructor(data, borrow=True) x = gpuarray_shared_constructor(data, borrow=True)
...@@ -497,5 +496,6 @@ def test_Gpujoin_inplace(): ...@@ -497,5 +496,6 @@ def test_Gpujoin_inplace():
c = join(0, x, z) c = join(0, x, z)
f = theano.function([s], theano.Out(c, borrow=True)) f = theano.function([s], theano.Out(c, borrow=True))
assert x.get_value(borrow=True, return_internal_type=True) is f(0) if not isinstance(mode_with_gpu, theano.compile.DebugMode):
assert x.get_value(borrow=True, return_internal_type=True) is f(0)
assert np.allclose(f(0), [3, 4, 5]) assert np.allclose(f(0), [3, 4, 5])
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论