From c38315f88bfb941d3ff4b896cbc70b651c6dedb4 Mon Sep 17 00:00:00 2001
From: Luis Vega <vegaluisjose@users.noreply.github.com>
Date: Fri, 21 May 2021 00:11:53 -0700
Subject: [PATCH] [BYOC][Verilator] Skip mobilenet test if Verilator is not
 available (#8094)

* skip mobilenet test when verilator is not available

* add skipped to pytest

* add pytest
---
 tests/python/contrib/test_verilator/test_mobilenet.py     | 5 +++++
 tests/python/contrib/test_verilator/test_verilator_ops.py | 3 +++
 2 files changed, 8 insertions(+)

diff --git a/tests/python/contrib/test_verilator/test_mobilenet.py b/tests/python/contrib/test_verilator/test_mobilenet.py
index 8447f19141ce..983c5d983d47 100644
--- a/tests/python/contrib/test_verilator/test_mobilenet.py
+++ b/tests/python/contrib/test_verilator/test_mobilenet.py
@@ -21,10 +21,12 @@
 from tvm.contrib import graph_executor as runtime
 
 import os
+import pytest
 from PIL import Image
 import numpy as np
 
 from test_verilator.infrastructure import (
+    skip_test,
     compile_hardware,
     compiler_opts,
     offload,
@@ -213,6 +215,7 @@ def is_tflite_available():
         return False
 
 
+@pytest.mark.skipif(skip_test(), reason="Skip because Verilator codegen is not available")
 def tmobilenet(lanes):
     """Mobilenet test template.
     Paramters
@@ -220,6 +223,8 @@ def tmobilenet(lanes):
     lanes : Int
         The number of vector lanes.
     """
+    if skip_test():
+        return
     if not is_tflite_available():
         return
     model = get_mobilenet_model()
diff --git a/tests/python/contrib/test_verilator/test_verilator_ops.py b/tests/python/contrib/test_verilator/test_verilator_ops.py
index 19ed1f067fed..3b0eb3130c2e 100644
--- a/tests/python/contrib/test_verilator/test_verilator_ops.py
+++ b/tests/python/contrib/test_verilator/test_verilator_ops.py
@@ -20,6 +20,7 @@
 
 import tvm
 from tvm import relay
+import pytest
 
 from test_verilator.infrastructure import (
     skip_test,
@@ -138,6 +139,7 @@ def print_test_info(test, lanes, cycles):
     print("test:{} vector-lanes:{} number of cycles:{}".format(test, lanes, cycles))
 
 
+@pytest.mark.skipif(skip_test(), reason="Skip because Verilator codegen is not available")
 def tadd(lanes):
     """Print counter
 
@@ -158,6 +160,7 @@ def tadd(lanes):
     print_test_info("add", lanes, cycles)
 
 
+@pytest.mark.skipif(skip_test(), reason="Skip because Verilator codegen is not available")
 def tbias(lanes):
     """Print counter