From 8de5fadac22f632fb1216cb8643ede5d7b3713f7 Mon Sep 17 00:00:00 2001
From: Jehandad Khan <jahandad@gmail.com>
Date: Tue, 1 Oct 2024 17:39:12 +0000
Subject: [PATCH] archive wheels

---
 .github/workflows/rocm-ci.yaml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.github/workflows/rocm-ci.yaml b/.github/workflows/rocm-ci.yaml
index eefc072889c5..a2814cf6bc46 100644
--- a/.github/workflows/rocm-ci.yaml
+++ b/.github/workflows/rocm-ci.yaml
@@ -16,6 +16,11 @@ jobs:
         run: |
           ./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
+        with:
+          name: rocm_jax_r6_1_3_py3_10_id${{ github.run_id }}
+          path: ${{ github.workspace }}/wheelhouse/*.whl
 
       
 #       - name: Try rocm-smi