Skip to content

Commit 77852cd

Browse files
committed
Add instance type and region fallbacks for Lambda capacity issues
1 parent 7a2db65 commit 77852cd

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

.github/workflows/gpu-e2e.yml

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,12 @@ on:
44
workflow_dispatch:
55
inputs:
66
instance_type:
7-
description: 'Lambda instance type'
8-
default: 'gpu_1x_a10'
7+
description: 'Lambda instance type(s), comma-separated for fallback'
8+
default: 'gpu_1x_a10,gpu_1x_a100,gpu_1x_h100_pcie'
9+
type: string
10+
region:
11+
description: 'Lambda region(s), comma-separated for fallback'
12+
default: 'us-south-1,us-west-1,us-east-1'
913
type: string
1014
epochs:
1115
description: 'Number of training epochs'
@@ -24,7 +28,8 @@ jobs:
2428
LAMBDA_API_KEY: ${{ secrets.LAMBDA_API_KEY }}
2529
LAMBDA_SSH_PRIVATE_KEY: ${{ secrets.LAMBDA_SSH_PRIVATE_KEY }}
2630
with:
27-
instance_type: ${{ inputs.instance_type || 'gpu_1x_a10' }}
31+
instance_type: ${{ inputs.instance_type || 'gpu_1x_a10,gpu_1x_a100,gpu_1x_h100_pcie' }}
32+
region: ${{ inputs.region || 'us-south-1,us-west-1,us-east-1' }}
2833
debug: ${{ inputs.debug || 'false' }}
2934

3035
gpu-test:

0 commit comments

Comments
 (0)