From 73906b97875f095cb685eb1e0ba05e0bdfc4e64e Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Tue, 22 Feb 2022 18:45:17 +0900 Subject: [PATCH 1/2] Simple workaround for PyTorch symbol crash problem in meta schedule test --- tests/python/unittest/test_meta_schedule_tune_relay.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/python/unittest/test_meta_schedule_tune_relay.py b/tests/python/unittest/test_meta_schedule_tune_relay.py index 7e6f89dfb149..8b6416feef8f 100644 --- a/tests/python/unittest/test_meta_schedule_tune_relay.py +++ b/tests/python/unittest/test_meta_schedule_tune_relay.py @@ -21,6 +21,8 @@ import numpy as np from typing import Tuple, List +import torch + import tvm from tvm import relay from tvm.ir import IRModule From b495fd7a6012871b02854d9306ed32c04a8266a5 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Wed, 23 Feb 2022 03:55:38 +0900 Subject: [PATCH 2/2] workaround for CI --- tests/python/unittest/test_meta_schedule_tune_relay.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/python/unittest/test_meta_schedule_tune_relay.py b/tests/python/unittest/test_meta_schedule_tune_relay.py index 8b6416feef8f..56b6feae067b 100644 --- a/tests/python/unittest/test_meta_schedule_tune_relay.py +++ b/tests/python/unittest/test_meta_schedule_tune_relay.py @@ -21,7 +21,10 @@ import numpy as np from typing import Tuple, List -import torch +try: + import torch +except ModuleNotFoundError: + pass import tvm from tvm import relay