test with other windows nodes

pytorch · Nov 12, 2024 · 5b7352c · 5b7352c
1 parent a624610
commit 5b7352c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/build-test-tensorrt-windows.yml b/.github/workflows/build-test-tensorrt-windows.yml
@@ -50,7 +50,7 @@ jobs:
       - name: Substitute runner
         id: substitute
         run: |
-          echo matrix="$(echo '${{ needs.generate-tensorrt-matrix.outputs.matrix }}' | sed -e 's/windows.g4dn.xlarge/windows.g6.4xlarge.nvidia.gpu/g')" >> ${GITHUB_OUTPUT}
+          echo matrix="$(echo '${{ needs.generate-tensorrt-matrix.outputs.matrix }}' | sed -e 's/windows.g4dn.xlarge/windows.g5.4xlarge.nvidia.gpu/g')" >> ${GITHUB_OUTPUT}
 
   build:
     needs: substitute-runner
@@ -102,6 +102,9 @@ jobs:
         export CI_BUILD=1
         pushd .
         cd tests/modules
+        python -c "import torch; print(torch.cuda.is_available())"
+        python -c "import torch; print(torch.cuda.get_device_name(0))"
+        python -c "import torch; print(torch.cuda.get_device_properties(0))"
         python hub.py
         popd
         pushd .
@@ -321,5 +324,5 @@ jobs:
   #       popd
 
 concurrency:
-  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}
-  cancel-in-progress: true
+  group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref_name }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}-${{ inputs.job-name }}-2
+  cancel-in-progress: false
diff --git a/.github/workflows/windows-test.yml b/.github/workflows/windows-test.yml
@@ -55,7 +55,7 @@ jobs:
       PYTHONUTF8: 1
       DOWNLOAD_ARTIFACT_NAME: pytorch_tensorrt_${{ matrix.tensorrt.version }}_${{ matrix.python_version }}_${{ matrix.desired_cuda }}_win_amd64
     name: ${{ inputs.job-name }}-${{ matrix.tensorrt.version }}-${{ matrix.python_version }}-${{ matrix.desired_cuda }}
-    runs-on: windows.8xlarge.nvidia.gpu.nonephemeral
+    runs-on: windows.4xlarge.nvidia.gpu
     defaults:
       run:
         shell: bash -l {0}