Fix bugs in workflow YML files

Author: charleshofer

…thub/workflows/reusable/build-docker.yml .github/workflows/build-docker.yml.github/workflows/reusable/build-docker.yml renamed to .github/workflows/build-docker.yml .github/workflows/reusable/build-docker.yml renamed to .github/workflows/build-docker.yml

@@ -29,6 +29,7 @@ jobs:
       - uses: actions/checkout@v4
       - name: Download wheel artifacts
         uses: actions/download-artifacts@v4
+        with:
           name: ${{ inputs.artifact-prefix }}_r${{ inputs.rocm-version }}
           path: ./wheelhouse
       - name: Build JAX docker image

…thub/workflows/reusable/build-wheels.yml .github/workflows/build-wheels.yml.github/workflows/reusable/build-wheels.yml renamed to .github/workflows/build-wheels.yml .github/workflows/reusable/build-wheels.yml renamed to .github/workflows/build-wheels.yml

@@ -10,9 +10,12 @@ on:
         required: true
         type: string
       artifact-prefix:
+        required: false
         default: 'plugin_wheels'
+        type: string
 
- build-plugin-docker:
+jobs:
+  build-plugin-docker:
     runs-on:  mi-250
     steps:
       - name: Clean up old runs

.github/workflows/ci.yml

@@ -22,7 +22,7 @@ jobs:
     strategy:
       matrix:
         rocm-version: ["6.3.3"]
-    uses: jax/rocm-jax/.github/workflows/reuseable/build-wheels.yml@master
+    uses: rocm/rocm-jax/.github/workflows/build-wheels.yml@master
     with:
       python-versions: ${{ env.PYTHON_VERSIONS }}
       rocm-version: ${{ matrix.rocm-version }}
@@ -31,7 +31,7 @@ jobs:
       matrix:
         python-version: ["3.10.13,3.12.8"]
         rocm-version: ["6.3.3"]
-    uses: jax/rocm-jax/.github/workflows/reusable/build-docker.yml@master
+    uses: rocm/rocm-jax/.github/workflows/build-docker.yml@master
     with:
       rocm-version: ${{ matrix.python-version }}
   run-python-unit-tests: