This is an automated email from the ASF dual-hosted git repository.
marcoabreu pushed a commit to branch devel-arm
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 6bd37f9b60b74c7b220d30bd909f0044234bab53
Author: Pedro Larroy <pedro.lar...@gmail.com>
AuthorDate: Wed Mar 28 17:45:18 2018 +0200
Reduce memory usage on ndarray tests (topk)
---
tests/python/unittest/test_ndarray.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/tests/python/unittest/test_ndarray.py
b/tests/python/unittest/test_ndarray.py
index a060465..7c50a42 100644
--- a/tests/python/unittest/test_ndarray.py
+++ b/tests/python/unittest/test_ndarray.py
@@ -711,9 +711,8 @@ def test_order():
k=dat_size*dat_size*dat_size*dat_size, is_ascend=False)
assert_almost_equal(nd_ret_argsort, gt)
- # test topk with a big shape
- a = mx.nd.arange(0, 54686454, step=1, repeat=1)
- assert_almost_equal(a.topk(k=54686454).asnumpy(), a.asnumpy()[::-1])
+ a = mx.nd.arange(0, 1024, step=1, repeat=1)
+ assert_almost_equal(a.topk(k=1024).asnumpy(), a.asnumpy()[::-1])
# Repeat those tests that don't involve indices. These should pass even
with
# duplicated input data values (over many repeated runs with different
random seeds,
--
To stop receiving notification emails like this one, please contact
marcoab...@apache.org.
