[cleanup] Use `"` instead of `'` for string literals

This is the more predominant style in the codebase. I'm sure there are
more in other parts of the codebase but it's hard to search/replace.
pull/1715/head snapshot-20221212.685
Sean Silva 2022-12-09 13:13:15 +00:00
parent d4862ec611
commit a595942033
1 changed files with 32 additions and 32 deletions

View File

@ -34,37 +34,37 @@ from torch_mlir_e2e_test.test_suite import register_all_tests
register_all_tests() register_all_tests()
def _get_argparse(): def _get_argparse():
config_choices = ['native_torch', 'torchscript', 'linalg', 'mhlo', 'tosa', 'lazy_tensor_core', 'torchdynamo'] config_choices = ["native_torch", "torchscript", "linalg", "mhlo", "tosa", "lazy_tensor_core", "torchdynamo"]
parser = argparse.ArgumentParser(description='Run torchscript e2e tests.') parser = argparse.ArgumentParser(description="Run torchscript e2e tests.")
parser.add_argument('-c', '--config', parser.add_argument("-c", "--config",
choices=config_choices, choices=config_choices,
default='linalg', default="linalg",
help=f''' help=f"""
Meaning of options: Meaning of options:
"linalg": run through torch-mlir's default Linalg-on-Tensors backend. "linalg": run through torch-mlir"s default Linalg-on-Tensors backend.
"mhlo": run through torch-mlir's default MHLO backend. "mhlo": run through torch-mlir"s default MHLO backend.
"tosa": run through torch-mlir's default TOSA backend. "tosa": run through torch-mlir"s default TOSA backend.
"native_torch": run the torch.nn.Module as-is without compiling (useful for verifying model is deterministic; ALL tests should pass in this configuration). "native_torch": run the torch.nn.Module as-is without compiling (useful for verifying model is deterministic; ALL tests should pass in this configuration).
"torchscript": compile the model to a torch.jit.ScriptModule, and then run that as-is (useful for verifying TorchScript is modeling the program correctly). "torchscript": compile the model to a torch.jit.ScriptModule, and then run that as-is (useful for verifying TorchScript is modeling the program correctly).
"lazy_tensor_core": run the model through the Lazy Tensor Core frontend and execute the traced graph. "lazy_tensor_core": run the model through the Lazy Tensor Core frontend and execute the traced graph.
"torchdynamo": run the model through the TorchDynamo frontend and execute the graph using Linalg-on-Tensors. "torchdynamo": run the model through the TorchDynamo frontend and execute the graph using Linalg-on-Tensors.
''') """)
parser.add_argument('-f', '--filter', default='.*', help=''' parser.add_argument("-f", "--filter", default=".*", help="""
Regular expression specifying which tests to include in this run. Regular expression specifying which tests to include in this run.
''') """)
parser.add_argument('-v', '--verbose', parser.add_argument("-v", "--verbose",
default=False, default=False,
action='store_true', action="store_true",
help='report test results with additional detail') help="report test results with additional detail")
parser.add_argument('-s', '--sequential', parser.add_argument("-s", "--sequential",
default=False, default=False,
action='store_true', action="store_true",
help='''Run tests sequentially rather than in parallel. help="""Run tests sequentially rather than in parallel.
This can be useful for debugging, since it runs the tests in the same process, This can be useful for debugging, since it runs the tests in the same process,
which make it easier to attach a debugger or get a stack trace.''') which make it easier to attach a debugger or get a stack trace.""")
parser.add_argument('--crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed', parser.add_argument("--crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed",
metavar="TEST", type=str, nargs='+', metavar="TEST", type=str, nargs="+",
help='A set of tests to not attempt to run, since they crash and cannot be XFAILed.') help="A set of tests to not attempt to run, since they crash and cannot be XFAILed.")
return parser return parser
def main(): def main():
@ -74,25 +74,25 @@ def main():
test.unique_name for test in GLOBAL_TEST_REGISTRY) test.unique_name for test in GLOBAL_TEST_REGISTRY)
# Find the selected config. # Find the selected config.
if args.config == 'linalg': if args.config == "linalg":
config = LinalgOnTensorsBackendTestConfig(RefBackendLinalgOnTensorsBackend()) config = LinalgOnTensorsBackendTestConfig(RefBackendLinalgOnTensorsBackend())
xfail_set = LINALG_XFAIL_SET xfail_set = LINALG_XFAIL_SET
if args.config == 'tosa': if args.config == "tosa":
config = TosaBackendTestConfig(LinalgOnTensorsTosaBackend()) config = TosaBackendTestConfig(LinalgOnTensorsTosaBackend())
xfail_set = all_test_unique_names - TOSA_PASS_SET xfail_set = all_test_unique_names - TOSA_PASS_SET
if args.config == 'mhlo': if args.config == "mhlo":
config = MhloBackendTestConfig(LinalgOnTensorsMhloBackend()) config = MhloBackendTestConfig(LinalgOnTensorsMhloBackend())
xfail_set = all_test_unique_names - MHLO_PASS_SET xfail_set = all_test_unique_names - MHLO_PASS_SET
elif args.config == 'native_torch': elif args.config == "native_torch":
config = NativeTorchTestConfig() config = NativeTorchTestConfig()
xfail_set = {} xfail_set = {}
elif args.config == 'torchscript': elif args.config == "torchscript":
config = TorchScriptTestConfig() config = TorchScriptTestConfig()
xfail_set = {} xfail_set = {}
elif args.config == 'lazy_tensor_core': elif args.config == "lazy_tensor_core":
config = LazyTensorCoreTestConfig() config = LazyTensorCoreTestConfig()
xfail_set = LTC_XFAIL_SET xfail_set = LTC_XFAIL_SET
elif args.config == 'torchdynamo': elif args.config == "torchdynamo":
config = TorchDynamoTestConfig() config = TorchDynamoTestConfig()
xfail_set = TORCHDYNAMO_XFAIL_SET xfail_set = TORCHDYNAMO_XFAIL_SET
@ -101,7 +101,7 @@ def main():
if args.crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed is not None: if args.crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed is not None:
for arg in args.crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed: for arg in args.crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed:
if arg not in all_test_unique_names: if arg not in all_test_unique_names:
print(f'ERROR: --crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed argument "{arg}" is not a valid test name') print(f"ERROR: --crashing_tests_to_not_attempt_to_run_and_a_bug_is_filed argument '{arg}' is not a valid test name")
sys.exit(1) sys.exit(1)
# Find the selected tests, and emit a diagnostic if none are found. # Find the selected tests, and emit a diagnostic if none are found.
@ -111,9 +111,9 @@ def main():
] ]
if len(tests) == 0: if len(tests) == 0:
print( print(
f'ERROR: the provided filter {args.filter!r} does not match any tests' f"ERROR: the provided filter {args.filter!r} does not match any tests"
) )
print('The available tests are:') print("The available tests are:")
for test in available_tests: for test in available_tests:
print(test.unique_name) print(test.unique_name)
sys.exit(1) sys.exit(1)
@ -132,6 +132,6 @@ def _suppress_warnings():
warnings.filterwarnings("ignore", warnings.filterwarnings("ignore",
message="A builtin ctypes object gave a PEP3118 format string that does not match its itemsize") message="A builtin ctypes object gave a PEP3118 format string that does not match its itemsize")
if __name__ == '__main__': if __name__ == "__main__":
_suppress_warnings() _suppress_warnings()
main() main()