test

huggingface · Aug 30, 2024 · b13d7e9 · b13d7e9
1 parent 9d6e0d4
commit b13d7e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -40,12 +40,12 @@ jobs:
       - name: Set up ROCM_DOCKER_DEVICES env var
         run: |
           chmod +x scripts/rocm_docker_devices.sh
-          scripts/rocm_docker_devices.sh
-        shell: bash
+          ./scripts/rocm_docker_devices.sh
 
       - name: Set outputs
+        # use script to get the devices
         id: rocm_docker_devices
-        run: echo "rocm_docker_devices={$ROCM_DOCKER_DEVICES}" >> $GITHUB_OUTPUT
+        run: echo "rocm_docker_devices=$(./scripts/rocm_docker_devices.sh)" >> $GITHUB_ENV
 
       - name: Print outputs
         run: echo "${{ steps.rocm_docker_devices.outputs.rocm_docker_devices }}"

diff --git a/scripts/rocm_docker_devices.sh b/scripts/rocm_docker_devices.sh
@@ -23,7 +23,4 @@ for INDEX in "${DEVICE_INDICES[@]}"; do
   fi
 done
 
-echo "ROCM_DOCKER_DEVICES=$ROCM_DOCKER_DEVICES"
-
-# export the ROCM_DOCKER_DEVICES variable
-export ROCM_DOCKER_DEVICES=$ROCM_DOCKER_DEVICES
+echo $ROCM_DOCKER_DEVICES