Disable timeout 1200sec issue for large tensor tests (apache#18201)

AntiZpvoh · Jul 6, 2020 · b24b684 · b24b684
1 parent 2a48960
commit b24b684
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py
@@ -29,6 +29,8 @@
 from mxnet import gluon, nd
 from common import with_seed
 import unittest
+import pytest
+
 
 # dimension constants
 MEDIUM_X = 10000
@@ -41,6 +43,7 @@
 RNN_LARGE_TENSOR = 2**28
 
 
+@pytest.mark.timeout(0)
 def test_nn():
     def check_gluon_embedding():
         m = gluon.nn.Embedding(SMALL_Y, MEDIUM_X)
@@ -591,6 +594,7 @@ def check_rnn():
     check_rnn()
 
 
+@pytest.mark.timeout(0)
 def test_tensor():
     def check_ndarray_zeros():
         a = nd.zeros(shape=(LARGE_X, SMALL_Y))
@@ -1196,6 +1200,7 @@ def check_correctness(mxnet_op, numpy_op, atol=1e-3):
     check_binary_broadcast()
 
 
+@pytest.mark.timeout(0)
 def test_basic():
     def check_elementwise():
         a = nd.ones(shape=(LARGE_X, SMALL_Y))
@@ -1810,6 +1815,7 @@ def check_minimum():
     check_minimum()
 
 
+@pytest.mark.timeout(0)
 def test_sparse_dot():
     shape = (2, VLARGE_X)
     sp_mat1 = nd.sparse.csr_matrix(([2], [6], [0, 1, 1]), shape=shape)

diff --git a/tests/nightly/test_large_vector.py b/tests/nightly/test_large_vector.py
@@ -29,12 +29,15 @@
 from mxnet import gluon, nd
 from common import with_seed
 import unittest
+import pytest
+
 
 # dimension constants
 LARGE_X = 4300000000
 MEDIUM_X = 1000000000
 
 
+@pytest.mark.timeout(0)
 def test_nn():
     def check_dense():
         data = mx.nd.ones(shape=LARGE_X)
@@ -161,6 +164,7 @@ def check_sequence_last():
     check_sequence_reverse()
 
 
+@pytest.mark.timeout(0)
 def test_tensor():
     def check_ndarray_zeros():
         a = nd.zeros(shape=LARGE_X)
@@ -462,6 +466,7 @@ def check_correctness(mxnet_op, numpy_op, atol=1e-3):
     check_binary_broadcast()
 
 
+@pytest.mark.timeout(0)
 def test_basic():
     def check_elementwise():
         a = nd.ones(shape=LARGE_X)