Allow each model to choose its own runner (#277)

Summary:
It's a good idea to allow each model to choose its own runner.  This also help unblock #212 until we know the root cause of why it runs OOM.

Pull Request resolved: #277

Reviewed By: kirklandsign

Differential Revision: D49176509

Pulled By: huydhn

fbshipit-source-id: b5e74943fb62839e3b9a7ff132842c8614d0f84b

Author: huydhn

.ci/scripts/gather_test_models.py

@@ -16,6 +16,11 @@
     "buck2",
     "cmake",
 ]
+DEFAULT_RUNNER = "linux.2xlarge"
+RUNNERS = {
+    # This one runs OOM on smaller runner, the root cause is unclear (T163016365)
+    "w2l": "linux.12xlarge"
+}
 
 
 def set_output(name: str, val: Any) -> None:
@@ -53,6 +58,7 @@ def export_models_for_ci() -> None:
                     "model": name,
                     "quantization": quantization,
                     "xnnpack_delegation": xnnpack_delegation,
+                    "runner": RUNNERS.get(name, DEFAULT_RUNNER),
                 }
             )
     set_output("models", json.dumps(models))

.github/workflows/pull.yml

@@ -43,7 +43,7 @@ jobs:
       matrix: ${{ fromJSON(needs.gather-models.outputs.models) }}
       fail-fast: false
     with:
-      runner: linux.2xlarge
+      runner: ${{ matrix.runner }}
       docker-image: executorch-ubuntu-22.04-clang12
       submodules: 'true'
       ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
@@ -99,7 +99,7 @@ jobs:
           - build-tool: cmake
       fail-fast: false
     with:
-      runner: linux.2xlarge
+      runner: ${{ matrix.runner }}
       docker-image: executorch-ubuntu-22.04-clang12
       submodules: 'true'
       ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
@@ -148,7 +148,7 @@ jobs:
           - build-tool: cmake
       fail-fast: false
     with:
-      runner: linux.2xlarge
+      runner: ${{ matrix.runner }}
       docker-image: executorch-ubuntu-22.04-clang12
       submodules: 'true'
       ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}