Refs/heads/add nightly test multi gpu configs (#12870)

This commit is contained in:
alisonshao
2025-11-08 15:14:50 -08:00
committed by GitHub
parent 5f02b918ec
commit d3a03aeef8
3 changed files with 59 additions and 12 deletions
+7 -1
View File
@@ -645,6 +645,12 @@ if __name__ == "__main__":
type=int,
help="Use auto load balancing. The number of parts.",
)
arg_parser.add_argument(
"--continue-on-error",
action="store_true",
default=False,
help="Continue running remaining tests even if one fails (useful for nightly tests)",
)
args = arg_parser.parse_args()
print(f"{args=}")
@@ -662,5 +668,5 @@ if __name__ == "__main__":
print("The running tests are ", [f.name for f in files])
exit_code = run_unittest_files(files, args.timeout_per_file)
exit_code = run_unittest_files(files, args.timeout_per_file, args.continue_on_error)
exit(exit_code)