Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
skipping flaky tests randint,log_softmax. Skipped topk due to increas…
Browse files Browse the repository at this point in the history
…ed memory footprint (#17410)
  • Loading branch information
access2rohit authored and apeforest committed Jan 23, 2020
1 parent ba47372 commit b72d195
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
6 changes: 5 additions & 1 deletion tests/nightly/test_large_array.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,9 @@

from mxnet.test_utils import rand_ndarray, assert_almost_equal, rand_coord_2d, default_context, check_symbolic_forward, create_2d_tensor
from mxnet import gluon, nd
from tests.python.unittest.common import with_seed, with_post_test_cleanup
from tests.python.unittest.common import with_seed, with_post_test_cleanup, teardown
from nose.tools import with_setup
import unittest

# dimension constants
MEDIUM_X = 10000
Expand Down Expand Up @@ -128,6 +129,7 @@ def np_softmax(x, axis=-1, temperature=1.0):
x /= np.sum(x, axis=axis, keepdims=True)
return x

@unittest.skip("log_softmax flaky, tracked at https://github.com/apache/incubator-mxnet/issues/17397")
def check_log_softmax():
ndim = 2
shape = (SMALL_Y, LARGE_X)
Expand Down Expand Up @@ -474,6 +476,7 @@ def check_ndarray_random_uniform():
a = nd.random.uniform(shape=(LARGE_X, SMALL_Y))
assert a[-1][0] != 0

@unittest.skip("Randint flaky, tracked at https://github.com/apache/incubator-mxnet/issues/16172")
@with_seed()
def check_ndarray_random_randint():
a = nd.random.randint(100, 10000, shape=(LARGE_X, SMALL_Y))
Expand Down Expand Up @@ -1089,6 +1092,7 @@ def check_sort():
s = nd.sort(b, is_ascend=False)
assert np.sum(s[0].asnumpy() == 0).all()

@unittest.skip("Topk takes lot of memory!, tracked at https://github.com/apache/incubator-mxnet/issues/17411")
def check_topk():
b = create_2d_tensor(rows=LARGE_X, columns=SMALL_Y)
k = nd.topk(b, k=10, axis=0, dtype=np.int64)
Expand Down
2 changes: 1 addition & 1 deletion tests/nightly/test_large_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@

from mxnet.test_utils import rand_ndarray, assert_almost_equal, rand_coord_2d, create_vector
from mxnet import gluon, nd
from tests.python.unittest.common import with_seed
from tests.python.unittest.common import with_seed, teardown
from nose.tools import with_setup
import unittest

Expand Down

0 comments on commit b72d195

Please sign in to comment.