commit f81a36bd70f2c3abd921c921b1121e7c3f52d52c
author Geoffrey Martin-Noble <gcmn@google.com> Mon Jun 07 21:35:14 2021 -0700
committer GitHub <noreply@github.com> Mon Jun 07 21:35:14 2021 -0700
tree 917271e98678b5237f5916e73e08fc2c4da3a787
parent 02c7a8a8a7847fe20cd5b12fc4766e0ed03ac88d

Extend timeout for vulkan model tests (#6138)

The mobile bert fake weights test times out about 3% of the time
internally.

build_tools/bazel/iree_check_test.bzl

diff --git a/build_tools/bazel/iree_check_test.bzl b/build_tools/bazel/iree_check_test.bzl
index bebe02f..04d3afe 100644
--- a/build_tools/bazel/iree_check_test.bzl
+++ b/build_tools/bazel/iree_check_test.bzl
@@ -23,6 +23,7 @@
         compiler_flags = [],
         runner_args = [],
         tags = [],
+        timeout = None,
         **kwargs):
     """Creates an iree-check-module test for the specified source file.
 
@@ -37,6 +38,7 @@
           are passed automatically.
       tags: additional tags to apply to the generated test. A tag "driver=DRIVER" is added
           automatically.
+      timeout: timeout for the generated tests.
       **kwargs: any additional attributes to pass to the underlying run_binary_test.
     """
     bytecode_module_name = name + "_bytecode_module"
@@ -60,6 +62,7 @@
         data = [":%s" % bytecode_module_name],
         test_binary = "//iree/tools:iree-check-module",
         tags = tags + ["driver=%s" % driver],
+        timeout = timeout,
         **kwargs
     )
 
@@ -71,6 +74,7 @@
         compiler_flags = [],
         runner_args = [],
         tags = [],
+        timeout = None,
         **kwargs):
     """Creates a test suite of iree-check-module tests for a single backend/driver pair.
 
@@ -88,6 +92,7 @@
           create a separate suite or iree_check_test.
       tags: tags to apply to the generated tests. Note that as in standard test suites, manual
           is treated specially and will also apply to the test suite itself.
+      timeout: timeout for the generated tests.
       **kwargs: any additional attributes to pass to the underlying tests and test suite.
     """
     tests = []
@@ -101,6 +106,7 @@
             compiler_flags = compiler_flags,
             runner_args = runner_args,
             tags = tags,
+            timeout = timeout,
             **kwargs
         )
         tests.append(test_name)