add pyteset decorator to pass CI

apache · Jan 27, 2022 · b0b24e3 · b0b24e3
1 parent de94153
commit b0b24e3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/python/unittest/test_meta_schedule_byoc_tensorrt.py b/tests/python/unittest/test_meta_schedule_byoc_tensorrt.py
@@ -201,6 +201,9 @@ def eval_func(rt_mod, device, evaluator_config, repeated_args):
                 ).evaluate()
 
 
+@tvm.testing.requires_cuda
+@has_tensorrt_codegen
+@has_tensorrt_runtime
 def test_conv2d_relu():
     data_shape = (1, 1280, 14, 14)
     out_channels = 256