mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-14 19:06:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Python and ROCm version matrix for rocm-main (#314)

Browse Source
This commit is contained in:
charleshofer 2025-04-02 11:49:03 -05:00 committed by GitHub
parent 7b20b7c0b7
commit 1f93b4b9b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
.github/workflows/rocm-ci.yml vendored
View File

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