Skip to content

Commit b75207f

Browse files
authored
Only add symm memory code path for auto labeling of h100-distributed (#6864)
https://github.com/pytorch/test-infra/pull/6833/files gets reverted because we enabled the auto labeling for too broad code path, so we want to narrow down to symmetric memory only PRs.
1 parent ebcaaf5 commit b75207f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchci/lib/bot/autoLabelBot.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,9 @@ const repoSpecificAutoLabels: { [repo: string]: [RegExp, string][] } = {
150150
[/aten\/src\/ATen\/mps/gi, "ciflow/mps"],
151151
[/aten\/src\/ATen\/native\/mps/gi, "ciflow/mps"],
152152
[/torch\/_inductor\/codegen\/mps.py/gi, "ciflow/mps"],
153+
[/torch\/csrc\/distributed\/c10d\/symm_mem/gi, "ciflow/h100-symm-mem"],
154+
[/torch\/distributed\/_symmetric_memory/gi, "ciflow/h100-symm-mem"],
155+
[/test\/distributed\/.*mem.*/gi, "ciflow/h100-symm-mem"],
153156
[/test\/test_mps.py/gi, "ciflow/mps"],
154157
[/test\/inductor\/test_mps_basic.py/gi, "ciflow/mps"],
155158
],

0 commit comments

Comments
 (0)