@@ -16,12 +16,21 @@ jobs:
16
16
strategy :
17
17
matrix :
18
18
include :
19
- - { shard: 1, num_shards: 6, runner: "linux.16xlarge.nvidia.gpu" }
20
- - { shard: 2, num_shards: 6, runner: "linux.g5.4xlarge.nvidia.gpu" }
21
- - { shard: 3, num_shards: 6, runner: "linux.4xlarge.nvidia.gpu" }
22
- - { shard: 4, num_shards: 6, runner: "linux.4xlarge.nvidia.gpu" }
23
- - { shard: 5, num_shards: 6, runner: "linux.4xlarge.nvidia.gpu" }
24
- - { shard: 6, num_shards: 6, 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.g5.4xlarge.nvidia.gpu" }
22
+ - { shard: 4, num_shards: 15, runner: "linux.g5.4xlarge.nvidia.gpu" }
23
+ - { shard: 5, num_shards: 15, runner: "linux.g5.4xlarge.nvidia.gpu" }
24
+ - { shard: 6, num_shards: 15, runner: "linux.g5.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" }
25
34
fail-fast : false
26
35
runs-on : ${{ matrix.runner }}
27
36
env :
@@ -145,7 +154,7 @@ jobs:
145
154
shell : bash
146
155
env :
147
156
DOCKER_IMAGE : ${{ steps.docker-image.outputs.docker-image }}
148
- NUM_WORKERS : 6
157
+ NUM_WORKERS : 15
149
158
WORKER_ID : ${{ matrix.shard }}
150
159
COMMIT_ID : ${{ github.sha }}
151
160
JOB_TYPE : manager
0 commit comments