[TEST][FLAKY] fix random fail

Author: FrozenGene

src/runtime/contrib/random/random.cc

@@ -123,5 +123,11 @@ TVM_REGISTER_GLOBAL("tvm.contrib.random.random_fill").set_body([](TVMArgs args,
   entry->random_engine.RandomFill(out);
 });
 
+TVM_REGISTER_GLOBAL("tvm.contrib.random.seed").set_body([](TVMArgs args, TVMRetValue* ret) {
+  RandomThreadLocalEntry* entry = RandomThreadLocalEntry::ThreadLocal();
+  uint64_t seed = args[0];
+  entry->random_engine.Seed(seed);
+});
+
 }  // namespace contrib
 }  // namespace tvm

tests/python/contrib/test_random.py

@@ -38,6 +38,11 @@ def enabled_ctx_list():
 def test_randint():
     m = 1024
     n = 1024
+    if not tvm.get_global_func("tvm.contrib.random.seed", True):
+        print("skip because extern function is not available")
+        return
+    seed = tvm.get_global_func("tvm.contrib.random.seed")
+    seed(0)
     A = random.randint(-127, 128, size=(m, n), dtype='int32')
     s = te.create_schedule(A.op)
 
@@ -62,6 +67,11 @@ def verify(target="llvm"):
 def test_uniform():
     m = 1024
     n = 1024
+    if not tvm.get_global_func("tvm.contrib.random.seed", True):
+        print("skip because extern function is not available")
+        return
+    seed = tvm.get_global_func("tvm.contrib.random.seed")
+    seed(0)
     A = random.uniform(0, 1, size=(m, n))
     s = te.create_schedule(A.op)
 
@@ -86,6 +96,11 @@ def verify(target="llvm"):
 def test_normal():
     m = 1024
     n = 1024
+    if not tvm.get_global_func("tvm.contrib.random.seed", True):
+        print("skip because extern function is not available")
+        return
+    seed = tvm.get_global_func("tvm.contrib.random.seed")
+    seed(0)
     A = random.normal(3, 4, size=(m, n))
     s = te.create_schedule(A.op)