提交 bf5bd3d5 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test use less memory

上级 98a6fbb9
...@@ -19,8 +19,14 @@ def test_add(): ...@@ -19,8 +19,14 @@ def test_add():
for shape in ((), (0,), (3,), (2,3), (1,10000000),(10,1000000), (100,100000), for shape in ((), (0,), (3,), (2,3), (1,10000000),(10,1000000), (100,100000),
(1000,10000),(10000,1000), (1000,10000),(10000,1000),
(4100,33,34),(33,4100,34),(33,34,4100), (4100,33,34),(33,4100,34),(33,34,4100),
(4100,33,34,6),(33,4100,34,6),(33,34,4100,6),(33,34,6,4100), (4100,33,3,6),(33,4100,3,6),(33,3,4100,6),(33,3,6,4100),
(33,34,35,36,37)): (4100,3,34,6),(3,4100,34,6),(3,34,4100,6),(3,34,6,4100),
(4100,3,4,36),(3,4100,4,36),(3,4,4100,36),(3,4,36,4100),
(3,34,35,36,37),
(33,34,3,36,37),
(33,34,35,36,3),
):
a0 = theano._asarray(numpy.random.rand(*shape), dtype='float32') a0 = theano._asarray(numpy.random.rand(*shape), dtype='float32')
a1 = a0.copy() a1 = a0.copy()
b0 = cuda_ndarray.CudaNdarray(a0) b0 = cuda_ndarray.CudaNdarray(a0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论