@@ -16,18 +16,21 @@ jobs:
16
16
strategy :
17
17
matrix :
18
18
include :
19
- - { shard: 1, num_shards: 12, runner: "linux.16xlarge.nvidia.gpu" }
20
- - { shard: 2, num_shards: 12, runner: "linux.g5.4xlarge.nvidia.gpu" }
21
- - { shard: 3, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
22
- - { shard: 4, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
23
- - { shard: 5, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
24
- - { shard: 6, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
25
- - { shard: 7, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
26
- - { shard: 8, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
27
- - { shard: 9, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
28
- - { shard: 10, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
29
- - { shard: 11, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
30
- - { shard: 12, num_shards: 12, runner: "linux.4xlarge.nvidia.gpu" }
19
+ - { shard: 1, num_shards: 15, runner: "linux.16xlarge.nvidia.gpu" }
20
+ - { shard: 2, num_shards: 15, runner: "linux.g5.4xlarge.nvidia.gpu" }
21
+ - { shard: 3, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
22
+ - { shard: 4, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
23
+ - { shard: 5, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
24
+ - { shard: 6, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
25
+ - { shard: 7, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
26
+ - { shard: 8, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
27
+ - { shard: 9, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
28
+ - { shard: 10, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
29
+ - { shard: 11, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
30
+ - { shard: 12, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
31
+ - { shard: 13, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
32
+ - { shard: 14, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
33
+ - { shard: 15, num_shards: 15, runner: "linux.4xlarge.nvidia.gpu" }
31
34
fail-fast : false
32
35
runs-on : ${{ matrix.runner }}
33
36
env :
@@ -150,7 +153,7 @@ jobs:
150
153
shell : bash
151
154
env :
152
155
DOCKER_IMAGE : ${{ steps.docker-image.outputs.docker-image }}
153
- NUM_WORKERS : 12
156
+ NUM_WORKERS : 15
154
157
WORKER_ID : ${{ matrix.shard }}
155
158
COMMIT_ID : ${{ github.sha }}
156
159
JOB_TYPE : manager
0 commit comments