cpu

Signed-off-by: Oliver Koenig <[email protected]>
NVIDIA · Oct 25, 2024 · de2dc61 · de2dc61
1 parent 8a1be96
commit de2dc61
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/.github/workflows/_test_template.yml b/.github/workflows/_test_template.yml
@@ -43,6 +43,7 @@ jobs:
     outputs:
       conclusion: ${{ steps.main.conclusion }}
       log: ${{ steps.main.outputs.log }}
+      potential_infra_failure: ${{ steps.main.outputs.potential_infra_failure }}
     steps:
         - name: Docker system cleanup
           run: |
@@ -54,7 +55,13 @@ jobs:
 
         - name: Start container
           run: |
-            docker run --rm -d --name nemo_container_${{ github.run_id }} --runtime=nvidia --gpus all --shm-size=64g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container_${{ github.run_id }} bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))"
+            if [[ "${{ inputs.RUNNER }}" == *cpu* ]]; then
+              ARG=("--runtime=nvidia --gpus all")
+            else
+              ARG=("")
+            fi
+
+            docker run --rm -d --name nemo_container_${{ github.run_id }} ${ARG[@]} --shm-size=64g --env TRANSFORMERS_OFFLINE=0 --env HYDRA_FULL_ERROR=1 --volume /mnt/datadrive/TestData:/home/TestData nemoci.azurecr.io/nemo_container_${{ github.run_id }} bash -c "sleep $(( ${{ inputs.TIMEOUT }} * 60 + 60 ))"
 
         - id: main
           name: Run main script