[Release v0.1.24] Cherry-pick 29 main PRs through #5952

.github/scripts/split_tests.sh

@@ -8,20 +8,23 @@
 # Parameters:
 #   --shards N     number of shards (required)
 #   --test-type TYPE test type, default aiter
+#   --select-file F  only shard test files listed in F (one per line)
 #   --dry-run      only output allocation plan, do not execute
 #   -v             Pytest's -v option, no effect
-# Exit code: always 0
+# Exit code: 0 on success, 1 for invalid arguments or selected paths
 
 set -euo pipefail
 
 SHARDS=0
 TEST_TYPE="aiter"
 DRY_RUN=0
+SELECT_FILE=""
 
 while [[ $# -gt 0 ]]; do
     case "$1" in
         --shards) SHARDS="$2"; shift 2 ;;
         --test-type) TEST_TYPE="$2"; shift 2 ;;
+        --select-file) SELECT_FILE="$2"; shift 2 ;;
         --dry-run) DRY_RUN=1; shift ;;
         -v|--verbose) shift ;; # compatibility, ignore
         *)
@@ -67,6 +70,42 @@ if [[ ${#ALL_FILES[@]} -eq 0 ]]; then
     exit 1
 fi
 
+# Apply the optional selection before sharding.
+if [[ -n "$SELECT_FILE" ]]; then
+    if [[ ! -f "$SELECT_FILE" ]]; then
+        echo "Selection file not found: $SELECT_FILE" >&2
+        exit 1
+    fi
+    declare -A SELECTED=()
+    while IFS= read -r line || [[ -n "$line" ]]; do
+        [[ -n "$line" ]] && SELECTED["$line"]=1
+    done < "$SELECT_FILE"
+    FILTERED=()
+    for f in "${ALL_FILES[@]}"; do
+        if [[ -n "${SELECTED[$f]:-}" ]]; then
+            FILTERED+=("$f")
+            unset "SELECTED[$f]"
+        fi
+    done
+    # Never silently drop a path the selector asked to run.
+    if [[ ${#SELECTED[@]} -gt 0 ]]; then
+        echo "Selection lists paths that are not test files under ${TEST_DIR}:" >&2
+        for f in "${!SELECTED[@]}"; do echo "  ${f}" >&2; done
+        exit 1
+    fi
+    echo "Test selection: ${#FILTERED[@]} of ${#ALL_FILES[@]} test files selected."
+    if [[ ${#FILTERED[@]} -eq 0 ]]; then
+        echo "Selection is empty — writing ${SHARDS} empty shard lists."
+        if [[ $DRY_RUN -eq 0 ]]; then
+            for ((s=0; s < SHARDS; s++)); do
+                : > "${TEST_TYPE}_shard_${s}.list"
+            done
+        fi
+        exit 0
+    fi
+    ALL_FILES=("${FILTERED[@]}")
+fi
+
 # ------------------------------
 # FILE_TIMES (seconds), unknown files default 15
 # ------------------------------

.github/workflows/aiter-test.yaml

@@ -376,7 +376,7 @@ jobs:
       contents: read
     steps:
       - name: Download all wheel artifacts
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         with:
           pattern: ${{ env.AITER_WHEEL_ARTIFACT_NAME }}-py*
           path: all-wheels
@@ -704,7 +704,7 @@ jobs:
 
       - name: Download test shard lists
         id: download_aiter_shards
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 15
         with:
           name: aiter_shards
@@ -732,7 +732,7 @@ jobs:
       - name: Download Aiter wheel artifact
         if: steps.restore_aiter_wheel.outputs.cache-hit != 'true'
         id: download_aiter_wheel
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         continue-on-error: true
         with:
@@ -742,7 +742,7 @@ jobs:
       - name: Retry Download Aiter wheel artifact
         if: steps.restore_aiter_wheel.outputs.cache-hit != 'true' && steps.download_aiter_wheel.outcome == 'failure'
         id: retry_aiter_wheel
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         with:
           name: ${{ env.AITER_WHEEL_ARTIFACT_NAME }}-py3.12
@@ -765,7 +765,7 @@ jobs:
       - name: Download Triton wheel artifact
         if: steps.restore_triton_wheel.outputs.cache-hit != 'true'
         id: download_triton_wheel
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         continue-on-error: true
         with:
@@ -988,7 +988,7 @@ jobs:
     needs: [detect_aiter_test_scope, standard]
     steps:
       - name: Download all test logs
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 15
         with:
           pattern: standard-test-log-*-shard-*
@@ -1053,7 +1053,7 @@ jobs:
       - name: Download Aiter wheel artifact
         if: steps.restore_aiter_wheel.outputs.cache-hit != 'true'
         id: download_aiter_wheel
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         continue-on-error: true
         with:
@@ -1062,7 +1062,7 @@ jobs:
 
       - name: Retry Download Aiter wheel artifact
         if: steps.restore_aiter_wheel.outputs.cache-hit != 'true' && steps.download_aiter_wheel.outcome == 'failure'
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         with:
           name: ${{ env.AITER_WHEEL_ARTIFACT_NAME }}-py3.12
@@ -1084,7 +1084,7 @@ jobs:
 
       - name: Download Triton wheel artifact
         if: steps.restore_triton_wheel.outputs.cache-hit != 'true'
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 30
         continue-on-error: true
         with:
@@ -1295,7 +1295,7 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Download standard test logs (mi35x only)
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8.0.1
         timeout-minutes: 15
         continue-on-error: true
         with:

.github/workflows/pr-title-tags.yaml

@@ -12,7 +12,8 @@ name: PR Title Tags & Labels
 # component namespace ([fix], [Perf], [MLA], [gfx1250], ...) are never
 # touched. The same rule applies to labels: only the component labels listed
 # in COMPONENT_LABELS are ever added or removed — human labels (ci:*, ...)
-# are left alone, while hand-toggling a managed component label is corrected
+# are left alone, except gfx942 config changes add ci:triton-300x.
+# Hand-toggling a managed component label is corrected
 # on the spot. To keep a PR's title untouched, add the 'no-auto-title' label —
 # it takes effect immediately, and removing it re-syncs the title on the spot.
 # Component labels are synced regardless of the opt-out: the AITER-triton
@@ -50,7 +51,8 @@ jobs:
           script: |
             // The component labels this workflow owns (name -> color). Only
             // these are ever added or removed; anything else on the PR
-            // (ci:*, no-auto-title, ...) belongs to humans.
+            // (ci:*, no-auto-title, ...) belongs to humans, except the
+            // add-only ci:triton-300x label for gfx942 config changes below.
             const COMPONENT_LABELS = {
               'Triton/Gluon': '1d76db',
               'ASM':          '5319e7',
@@ -86,12 +88,17 @@ jobs:
             const optedOut = pr.labels.some(l => l.name === 'no-auto-title');
 
             // ---- changed files (paginated) ----
-            const files = (await github.paginate(github.rest.pulls.listFiles, {
+            const changedFiles = await github.paginate(github.rest.pulls.listFiles, {
               owner: context.repo.owner,
               repo: context.repo.repo,
               pull_number: pr.number,
               per_page: 100,
-            })).map(f => f.filename);
+            });
+            if (changedFiles.length !== pr.changed_files) {
+              throw new Error(`Incomplete PR file list: ${changedFiles.length}/${pr.changed_files}`);
+            }
+            const files = changedFiles.map(f => f.filename);
+            const changedPaths = changedFiles.flatMap(f => [f.filename, f.previous_filename].filter(Boolean));
 
             // ---- path predicates ----
             const isDoc    = f => f.startsWith('docs/') || f.endsWith('.md');
@@ -218,3 +225,13 @@ jobs:
             } else {
               core.info('Labels already in sync.');
             }
+
+            if (changedPaths.some(f => f.startsWith('aiter/ops/triton/configs/gfx942/') && !isDoc(f))
+                && !pr.labels.some(l => l.name === 'ci:triton-300x')) {
+              await github.rest.issues.addLabels({
+                owner: context.repo.owner,
+                repo: context.repo.repo,
+                issue_number: pr.number,
+                labels: ['ci:triton-300x'],
+              });
+            }

.github/workflows/pr-welcome-comment.yaml

@@ -31,7 +31,8 @@ jobs:
             | Label | Tests |
             |-------|-------|
             | \`ci:gfx1250-ffm-triton\` | Run the five-shard gfx1250 FFM Triton test suite |
-            | \`ci:triton-300x\` | Run an additional Triton test job on MI300X in PRs; main branch always runs both MI35X and MI300X |
+            | \`ci:triton-300x\` | Run an additional Triton test job on MI300X in PRs (added automatically when gfx942 configs change); main branch always runs both MI35X and MI300X |
+            | \`ci:triton-355\` | Run the full Triton test suite on MI35X, not only the tests the change affects |
             | \`multigpu\` | Aiter multi-GPU tests on the 8-GPU runner |
             | \`ci:sglang\` | SGLang integration tests: DeepSeek-R1-MXFP4 accuracy, Qwen 3.5 accuracy |
             | \`ci:atom\` | ATOM benchmark: DeepSeek-R1-0528, GPT-OSS-120B |