Skip to content

Commit

Permalink
skipping randint flaky test for large vector and reordering op execut…
Browse files Browse the repository at this point in the history
…ion (apache#17388)
  • Loading branch information
access2rohit authored and Zheng committed Jan 21, 2020
1 parent 10c01de commit 96c84da
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 1 deletion.
10 changes: 10 additions & 0 deletions ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 3 additions & 1 deletion tests/nightly/test_large_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -147,14 +148,14 @@ 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()
check_layer_norm()
check_batchnorm()
check_sequence_mask()
check_sequence_reverse()
check_sequence_last()


def test_tensor():
Expand All @@ -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)
Expand Down

0 comments on commit 96c84da

Please sign in to comment.