From 7dc2004eabfc70156f4f90d63117b3e4ae0ce188 Mon Sep 17 00:00:00 2001 From: Ubuntu Date: Thu, 23 Jul 2020 22:52:41 +0000 Subject: [PATCH] passing mat size as arg --- python/mxnet/test_utils.py | 6 +++--- tests/nightly/test_large_array.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/mxnet/test_utils.py b/python/mxnet/test_utils.py index c2f4d7c923a8..c47952818081 100755 --- a/python/mxnet/test_utils.py +++ b/python/mxnet/test_utils.py @@ -298,9 +298,9 @@ def create_vector(size, dtype=np.int64): return a # For testing Large Square Matrix with total size > 2^32 elements -def get_large_identity_mat(): - A = nd.zeros((LARGE_SQ_X, LARGE_SQ_X)) - for i in range(LARGE_SQ_X): +def get_large_identity_mat(size): + A = nd.zeros((size, size)) + for i in range(size): A[i,i] = 1 return A diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py index 39236c53b10c..a3dd7d7eea6e 100644 --- a/tests/nightly/test_large_array.py +++ b/tests/nightly/test_large_array.py @@ -1204,7 +1204,7 @@ def run_trsm(inp): out = trsm(inp, inp) return inp.grad, out - A = get_large_identity_mat() + A = get_large_identity_mat(LARGE_SQ_X) grad, out = run_det(A) check_diag(grad, 0)