From 57cc953124b4d756fdfce82253337e472a6e2381 Mon Sep 17 00:00:00 2001 From: dsbarinov1 Date: Wed, 1 Feb 2023 12:08:02 +0300 Subject: [PATCH] Import threading + lint --- tests/python/contrib/test_random.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/python/contrib/test_random.py b/tests/python/contrib/test_random.py index 1b426bbbcb5c5..ddc06b07110eb 100644 --- a/tests/python/contrib/test_random.py +++ b/tests/python/contrib/test_random.py @@ -20,6 +20,7 @@ from tvm.contrib import random from tvm import rpc import tvm.testing +import threading def test_randint(): @@ -160,7 +161,7 @@ def test_random_fill_mt(): Particularly when MaxConcurrency != num_workers_used_ which is actual for big-little systems. """ no_exception_happened = True - + def test_body(): try: num_thread_used = 1 @@ -178,7 +179,7 @@ def test_body(): x = threading.Thread(target=test_body) x.start() x.join() - assert no_exception_happened + assert no_exception_happened if __name__ == "__main__":