File tree Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Original file line number Diff line number Diff line change 29
29
- uses : actions/checkout@v4
30
30
- name : Download wheel artifacts
31
31
uses : actions/download-artifacts@v4
32
+ with :
32
33
name : ${{ inputs.artifact-prefix }}_r${{ inputs.rocm-version }}
33
34
path : ./wheelhouse
34
35
- name : Build JAX docker image
Original file line number Diff line number Diff line change 10
10
required : true
11
11
type : string
12
12
artifact-prefix :
13
+ required : false
13
14
default : ' plugin_wheels'
15
+ type : string
14
16
15
- build-plugin-docker :
17
+ jobs :
18
+ build-plugin-docker :
16
19
runs-on : mi-250
17
20
steps :
18
21
- name : Clean up old runs
Original file line number Diff line number Diff line change 22
22
strategy :
23
23
matrix :
24
24
rocm-version : ["6.3.3"]
25
- uses : jax /rocm-jax/.github/workflows/reuseable /build-wheels.yml@master
25
+ uses : rocm /rocm-jax/.github/workflows/build-wheels.yml@master
26
26
with :
27
27
python-versions : ${{ env.PYTHON_VERSIONS }}
28
28
rocm-version : ${{ matrix.rocm-version }}
31
31
matrix :
32
32
python-version : ["3.10.13,3.12.8"]
33
33
rocm-version : ["6.3.3"]
34
- uses : jax /rocm-jax/.github/workflows/reusable /build-docker.yml@master
34
+ uses : rocm /rocm-jax/.github/workflows/build-docker.yml@master
35
35
with :
36
36
rocm-version : ${{ matrix.python-version }}
37
37
run-python-unit-tests :
You can’t perform that action at this time.
0 commit comments