Skip to content

Commit d0b7934

Browse files
authored
Allow triggering MLX CI w/ label (#19173)
Allow triggering MLX HF jobs from external contributors.
1 parent 35b9054 commit d0b7934

2 files changed

Lines changed: 20 additions & 0 deletions

File tree

.github/pytorch-probot.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ ciflow_push_tags:
66
- ciflow/cuda
77
- ciflow/cuda-perf
88
- ciflow/metal
9+
- ciflow/mlx
910
- ciflow/nightly
1011
- ciflow/trunk
1112
- ciflow/binaries

.github/workflows/mlx.yml

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@ on:
55
branches:
66
- main
77
- release/*
8+
tags:
9+
- ciflow/mlx/*
810
pull_request:
911
paths:
1012
- .github/workflows/mlx.yml
@@ -16,6 +18,10 @@ on:
1618
- examples/models/qwen3_5_moe/**
1719
workflow_dispatch:
1820

21+
concurrency:
22+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}
23+
cancel-in-progress: true
24+
1925
permissions: {}
2026

2127
jobs:
@@ -218,6 +224,10 @@ jobs:
218224
echo "::endgroup::"
219225
220226
test-mlx-voxtral:
227+
# Requires HuggingFace secrets — skip on fork PRs.
228+
# Maintainers can opt-in by applying the ciflow/mlx label, which
229+
# pushes a ciflow/mlx/<PR> tag that re-runs this workflow with secrets.
230+
if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name != 'pull_request'
221231
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
222232
secrets: inherit
223233
with:
@@ -275,6 +285,9 @@ jobs:
275285
echo "::endgroup::"
276286
277287
test-mlx-voxtral-realtime:
288+
# Requires HuggingFace secrets — skip on fork PRs.
289+
# Maintainers can opt-in by applying the ciflow/mlx label.
290+
if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name != 'pull_request'
278291
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
279292
secrets: inherit
280293
with:
@@ -347,6 +360,9 @@ jobs:
347360
echo "::endgroup::"
348361
349362
test-mlx-whisper:
363+
# Requires HuggingFace secrets — skip on fork PRs.
364+
# Maintainers can opt-in by applying the ciflow/mlx label.
365+
if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name != 'pull_request'
350366
uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
351367
secrets: inherit
352368
with:
@@ -458,6 +474,9 @@ jobs:
458474
echo "::endgroup::"
459475
460476
test-mlx-llm:
477+
# Requires HuggingFace secrets — skip on fork PRs.
478+
# Maintainers can opt-in by applying the ciflow/mlx label.
479+
if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name != 'pull_request'
461480
strategy:
462481
fail-fast: false
463482
matrix:

0 commit comments

Comments
 (0)