add test stage

ROCm · Oct 2, 2024 · 23dcab1 · 23dcab1
1 parent 4df858b
commit 23dcab1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.github/workflows/rocm-ci.yaml b/.github/workflows/rocm-ci.yaml
@@ -21,8 +21,6 @@ jobs:
         with:
           name: rocm_jax_r6_1_3_py3_10_id${{ github.run_id }}
           path: ${{ github.workspace }}/wheelhouse/*.whl
-
-
       - name: Detect GPUs
         run: |
           docker run \
@@ -32,3 +30,9 @@ jobs:
             --security-opt seccomp=unconfined \
             --shm-size=64G \
             rocm/rocm-terminal rocm-smi
+      - name: Run tests
+        env: 
+          DOCKER_IMG_NAME: rocm_jax_r6_1_3_py3_10_id${{ github.run_id }}
+        run: |
+          ./build/rocm/ci_build test ${DOCKER_IMG_NAME}
+