From d8a41ea09ee0268187f6b3aae7f3e3f783de33d2 Mon Sep 17 00:00:00 2001
From: Scott Todd <Scott.Todd@amd.com>
Date: Tue, 30 Jun 2026 13:10:53 -0700
Subject: [PATCH] Add python_version input to jax release workflow

---
 .github/workflows/multi_arch_release_linux_jax_wheels.yml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/multi_arch_release_linux_jax_wheels.yml b/.github/workflows/multi_arch_release_linux_jax_wheels.yml
index 52edefad..cace7672 100644
--- a/.github/workflows/multi_arch_release_linux_jax_wheels.yml
+++ b/.github/workflows/multi_arch_release_linux_jax_wheels.yml
@@ -34,6 +34,10 @@ on:
         description: "Repository to checkout. Defaults to ROCm/TheRock."
         type: string
         default: "ROCm/TheRock"
+      python_version:
+        description: "Single Python version to build (empty for full matrix)"
+        type: string
+        default: ''
 
 permissions:
   contents: read
@@ -53,3 +57,4 @@ jobs:
       tar_url: ${{ inputs.tar_url }}
       repository: ${{ inputs.repository || 'ROCm/TheRock' }}
       ref: ${{ inputs.ref || '' }}
+      python_version: ${{ inputs.python_version }}