[TEST][FLAKY] fix random fail (#6312)

* [TEST][FLAKY] fix random fail

* increase size and error check range

diff --git a/tests/python/contrib/test_random.py b/tests/python/contrib/test_random.py
index bc081a4..e61030b 100644 (file)
--- a/tests/python/contrib/test_random.py
+++ b/tests/python/contrib/test_random.py
@@ -36,8 +36,8 @@ def enabled_ctx_list():
 ENABLED_CTX_LIST = enabled_ctx_list()
 
 def test_randint():
-    m = 1024
-    n = 1024
+    m = 10240
+    n = 10240
     A = random.randint(-127, 128, size=(m, n), dtype='int32')
     s = te.create_schedule(A.op)
 
@@ -53,15 +53,15 @@ def test_randint():
         a = tvm.nd.array(np.zeros((m, n), dtype=A.dtype), ctx)
         f(a)
         na = a.asnumpy()
-        assert abs(np.mean(na)) < 0.2
+        assert abs(np.mean(na)) < 0.3
         assert np.min(na) == -127
         assert np.max(na) == 127
     verify()
 
 
 def test_uniform():
-    m = 1024
-    n = 1024
+    m = 10240
+    n = 10240
     A = random.uniform(0, 1, size=(m, n))
     s = te.create_schedule(A.op)
 
@@ -77,15 +77,15 @@ def test_uniform():
         a = tvm.nd.array(np.zeros((m, n), dtype=A.dtype), ctx)
         f(a)
         na = a.asnumpy()
-        assert abs(np.mean(na) - 0.5) < 1e-2
+        assert abs(np.mean(na) - 0.5) < 1e-1
         assert abs(np.min(na) - 0.0) < 1e-3
         assert abs(np.max(na) - 1.0) < 1e-3
     verify()
 
 
 def test_normal():
-    m = 1024
-    n = 1024
+    m = 10240
+    n = 10240
     A = random.normal(3, 4, size=(m, n))
     s = te.create_schedule(A.op)
 
@@ -101,7 +101,7 @@ def test_normal():
         a = tvm.nd.array(np.zeros((m, n), dtype=A.dtype), ctx)
         f(a)
         na = a.asnumpy()
-        assert abs(np.mean(na) - 3) < 1e-2
+        assert abs(np.mean(na) - 3) < 1e-1
         assert abs(np.std(na) - 4) < 1e-2
     verify()