Added uses_gpu tag.

apache · Oct 26, 2020 · 65454b7 · 65454b7
1 parent f3b613d
commit 65454b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -3841,10 +3841,10 @@ def test_loop():
     verify_count_loop()
 
 
+@tvm.testing.uses_gpu
 def test_if():
     # Given a bool scalar input cond.
     # return constant tensor x if cond is True, otherwise return constant tensor y.
-
     then_out = onnx.helper.make_tensor_value_info("then_out", onnx.TensorProto.FLOAT, [5])
     else_out = onnx.helper.make_tensor_value_info("else_out", onnx.TensorProto.FLOAT, [5])