mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +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
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