Skip to content

Commit 721eb20

Browse files
authored
Triaging: exclude TE when needed (#1764)
As it happened recently, avoid triaging TE library if `---exclude-transformer-engine` is given
1 parent 8b3c50b commit 721eb20

File tree

1 file changed

+8
-0
lines changed
  • .github/triage/jax_toolbox_triage

1 file changed

+8
-0
lines changed

.github/triage/jax_toolbox_triage/args.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,11 @@ def parse_args(args=None) -> argparse.Namespace:
260260
jax:https://<token>@host/repo.git,xla:https://<token>@host/repo.git
261261
""",
262262
)
263+
version_search_args.add_argument(
264+
"--exclude-transformer-engine",
265+
action="store_true",
266+
help="Exclude transformer-engine from the list of optional software to triage.",
267+
)
263268
parser.add_argument(
264269
"-v",
265270
"--container-mount",
@@ -375,4 +380,7 @@ def parse_args(args=None) -> argparse.Namespace:
375380
"--container must be passed for the container-level search"
376381
)
377382

383+
if args.exclude_transformer_engine:
384+
optional_software.remove("transformer-engine")
385+
378386
return args

0 commit comments

Comments
 (0)