From 1f93b4b9b59a9ede1db9a947b96470f1f55c9705 Mon Sep 17 00:00:00 2001
From: charleshofer <Charles.Hofer@amd.com>
Date: Wed, 2 Apr 2025 11:49:03 -0500
Subject: [PATCH] Add Python and ROCm version matrix for rocm-main (#314)

---
 .github/workflows/rocm-ci.yml | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/rocm-ci.yml b/.github/workflows/rocm-ci.yml
index 3eda5116d..4efbeee7c 100644
--- a/.github/workflows/rocm-ci.yml
+++ b/.github/workflows/rocm-ci.yml
@@ -17,13 +17,17 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  build-jax-in-docker: # strategy and matrix come here
+  build-jax-in-docker:
     runs-on:  mi-250
+    strategy:
+      matrix:
+        python: ["3.10.13"]
+        rocm: ["6.2.4", "6.3.3"]
     env:
       BASE_IMAGE: "ubuntu:22.04"
       TEST_IMAGE: ubuntu-jax-${{ github.run_id }}_${{ github.run_number }}_${{ github.run_attempt }}
-      PYTHON_VERSION: "3.10"
-      ROCM_VERSION: "6.2.4"
+      PYTHON_VERSION: ${{ matrix.python }}
+      ROCM_VERSION: ${{ matrix.rocm }}
       WORKSPACE_DIR: workdir_${{ github.run_id }}_${{ github.run_number }}_${{ github.run_attempt }}
     steps:
       - name: Clean up old runs