diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh index 60ab617c88ae..31d00d7eb8fa 100755 --- a/ci/docker/runtime_functions.sh +++ b/ci/docker/runtime_functions.sh @@ -1497,6 +1497,16 @@ nightly_test_large_vector() { nosetests-3.4 tests/nightly/test_large_vector.py:test_basic } +#Test Large Vectors +nightly_test_large_vector() { + set -ex + export PYTHONPATH=./python/ + export DMLC_LOG_STACK_TRACE_DEPTH=10 + nosetests-3.4 tests/nightly/test_large_vector.py:test_tensor + nosetests-3.4 tests/nightly/test_large_vector.py:test_nn + nosetests-3.4 tests/nightly/test_large_vector.py:test_basic +} + #Tests Amalgamation Build with 5 different sets of flags nightly_test_amalgamation() { set -ex diff --git a/tests/nightly/test_large_vector.py b/tests/nightly/test_large_vector.py index ceedab0ad5a3..6a55810613d4 100644 --- a/tests/nightly/test_large_vector.py +++ b/tests/nightly/test_large_vector.py @@ -25,6 +25,7 @@ from mxnet import gluon, nd from tests.python.unittest.common import with_seed from nose.tools import with_setup +import unittest # dimension constants LARGE_X = 4300000000 @@ -147,6 +148,7 @@ def check_sequence_last(): # check if it takes 2nd sequence from the first batch assert b[0] == a[1][0] + check_sequence_last() check_dense() check_regression() check_sign() @@ -154,7 +156,6 @@ def check_sequence_last(): check_batchnorm() check_sequence_mask() check_sequence_reverse() - check_sequence_last() def test_tensor(): @@ -178,6 +179,7 @@ def check_ndarray_random_uniform(): a = nd.random.uniform(shape=LARGE_X) assert a[-1] != 0 + @unittest.skip("Randint flaky, tracked at https://github.com/apache/incubator-mxnet/issues/16172") @with_seed() def check_ndarray_random_randint(): # check if randint can generate value greater than 2**32 (large)