diff --git a/.github/workflows/rocm-ci.yaml b/.github/workflows/rocm-ci.yaml
index a2814cf6bc46..5400daba43fa 100644
--- a/.github/workflows/rocm-ci.yaml
+++ b/.github/workflows/rocm-ci.yaml
@@ -17,7 +17,7 @@ jobs:
           ./build/rocm/ci_build.sh --rocm_version 6.1.3 \
           --keep_image --py_version 3.10 
       - name: Archive jax wheels
-        uses: actions/upload-artifact@v2
+        uses: actions/upload-artifact@v4
         with:
           name: rocm_jax_r6_1_3_py3_10_id${{ github.run_id }}
           path: ${{ github.workspace }}/wheelhouse/*.whl