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