Use OS runner labels in workflows (#3526)

This allows selecting, for example, a linux runner with `["linux", "a770"]` and a windows runner with `["windows", "a770"]`.

Author: pbchekin

.github/workflows/build-test-reusable.yml

@@ -74,7 +74,7 @@ env:
 jobs:
   integration-tests:
     name: Integration tests
-    runs-on: ${{ fromJson(inputs.runner_label && format('["{0}"]', inputs.runner_label) || format('["{0}", "{1}", "{2}"]', inputs.device, inputs.driver_version, inputs.runner_version)) }}
+    runs-on: ${{ fromJson(inputs.runner_label && format('["linux", "{0}"]', inputs.runner_label) || format('["linux", "{0}", "{1}", "{2}"]', inputs.device, inputs.driver_version, inputs.runner_version)) }}
     defaults:
       run:
         shell: bash -noprofile --norc -eo pipefail -c "source /opt/intel/oneapi/setvars.sh > /dev/null; source {0}"

.github/workflows/build-test-windows.yml

@@ -30,7 +30,9 @@ env:
 jobs:
   build:
     name: Build and test
-    runs-on: ${{ inputs.runner_label }}
+    runs-on:
+      - windows
+      - ${{ inputs.runner_label }}
     # Building PyTorch can take up to 4 hours on certain machines, increasing the timeout to 8 hours.
     timeout-minutes: 480
     steps:

.github/workflows/build-windows.yml

@@ -20,7 +20,9 @@ env:
 jobs:
   build:
     name: Build
-    runs-on: avc336
+    runs-on:
+      - windows
+      - avc336
     steps:
       - name: Enable long paths
         run: |

.github/workflows/conda-test-reusable.yml

@@ -54,6 +54,7 @@ jobs:
   run_tests:
     name: Test ${{ inputs.suite }} ${{ inputs.dtype }} ${{ inputs.mode }} ${{ inputs.test_mode }}
     runs-on:
+      - linux
       - ${{ inputs.runner_label || 'max1550' }}
     timeout-minutes: 720
     defaults:

.github/workflows/conda-test.yml

@@ -34,6 +34,7 @@ jobs:
   build:
     name: Test
     runs-on:
+      - linux
       - ${{ inputs.runner_label || 'rolling' }}
     timeout-minutes: 720
     defaults:

.github/workflows/e2e-reusable.yml

@@ -42,7 +42,9 @@ env:
 jobs:
   build:
     name: Build and test
-    runs-on: ${{ inputs.runner_label || 'win-a770' }}
+    runs-on:
+      - windows
+      - ${{ inputs.runner_label || 'win-a770' }}
     steps:
       - name: Enable long paths
         run: |

.github/workflows/inductor-tests-reusable.yml

@@ -57,6 +57,7 @@ jobs:
   build:
     name: Triton benchmarks
     runs-on:
+      - linux
       - ${{ inputs.runner_label || 'max1550' }}
     timeout-minutes: 720
     defaults:

.github/workflows/pip-test-windows.yml

@@ -128,4 +128,4 @@ jobs:
       dtype: ${{ matrix.dtype }}
       models: subset
       only_one_model: ""
-      runner_label: "${{ matrix.runner_label }}"
+      runner_label: ${{ inputs.runner_label }}