From ea6245f788fc05dec19807cafc975bfcefb09826 Mon Sep 17 00:00:00 2001 From: Luke Hutton Date: Thu, 29 Jun 2023 09:22:25 +0000 Subject: [PATCH] [AOT] Remove workaround to help resolve test flakiness Now that #10300 and #10314 have been closed, we should be able to remove an old previous attempt to help resolve test flakiness. Change-Id: I70c09d6ba5ffc5cb15d0b775732bd048b7ebfbb4 --- python/tvm/testing/aot.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/python/tvm/testing/aot.py b/python/tvm/testing/aot.py index b2814aff2d77..9ee3a84c8a38 100644 --- a/python/tvm/testing/aot.py +++ b/python/tvm/testing/aot.py @@ -17,7 +17,6 @@ # pylint: disable=use-list-literal, consider-using-with, f-string-without-interpolation """Common functions for AOT test cases""" import contextlib -import sys import datetime import os import pathlib @@ -830,14 +829,7 @@ def run_and_check_body(base_path): if verbose: print("Run command:\n", run_command) - # TODO(lhutton1) This is a quick and dirty work around to help temporarily reduce - # the flakyness of the tests. Will remove once #10300 and #10314 are resolved. - try: - _subprocess_check_log_output(run_command, build_path, run_log_path) - except RuntimeError as err: - print("Failed to run the module, having a second attempt...", file=sys.stderr) - print(err, file=sys.stderr) - _subprocess_check_log_output(run_command, build_path, run_log_path) + _subprocess_check_log_output(run_command, build_path, run_log_path) with open(run_log_path) as run_log: assert AOT_SUCCESS_TOKEN in run_log.read()