diff --git a/.github/workflows/call_precommit.yml b/.github/workflows/call_precommit.yml index 6c2f58ae9ce..52c72e1309e 100644 --- a/.github/workflows/call_precommit.yml +++ b/.github/workflows/call_precommit.yml @@ -136,9 +136,10 @@ jobs: env: NUM_WORKERS: 4 + pytorch-cuda: timeout-minutes: 40 - runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4 + runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4-staging if: ${{ inputs.gpu_enabled == true }} defaults: run: diff --git a/.github/workflows/examples.yml b/.github/workflows/examples.yml index ad1706a1e97..514a6e666b6 100644 --- a/.github/workflows/examples.yml +++ b/.github/workflows/examples.yml @@ -79,7 +79,7 @@ jobs: examples-cuda: name: Test examples CUDA [${{ matrix.group }}/1] - runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4 + runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4-staging timeout-minutes: 40 if: ${{ github.event_name != 'workflow_dispatch' || github.event.inputs.skip_gpu == 'false' }} strategy: diff --git a/.github/workflows/install.yml b/.github/workflows/install.yml index 6aca3518c1b..c5887eeb501 100644 --- a/.github/workflows/install.yml +++ b/.github/workflows/install.yml @@ -48,7 +48,7 @@ jobs: defaults: run: shell: bash - runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4 + runs-on: aks-linux-4-cores-28gb-gpu-tesla-t4-staging timeout-minutes: 20 env: DEBIAN_FRONTEND: noninteractive