Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion infra/marin-eu-west4-a.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ available_node_types:

tpu_slice_v6e_128:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v6e-128
runtimeVersion: v2-alpha-tpuv6e
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-eu-west4-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v5litepod-4
runtimeVersion: v2-alpha-tpuv5-lite
Expand Down
4 changes: 2 additions & 2 deletions infra/marin-eu-west4.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 4
min_workers: 2
node_config:
acceleratorType: v5litepod-4
runtimeVersion: v2-alpha-tpuv5-lite
Expand Down Expand Up @@ -194,7 +194,7 @@ available_node_types:

tpu_slice_v5e_128:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5litepod-128
runtimeVersion: v2-alpha-tpuv5-lite
Expand Down
4 changes: 2 additions & 2 deletions infra/marin-us-central1-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -141,7 +141,7 @@ available_node_types:

tpu_slice_v5p_8:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand Down
10 changes: 5 additions & 5 deletions infra/marin-us-central1.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -146,7 +146,7 @@ available_node_types:

tpu_slice_v5p_8:
max_workers: 1024
min_workers: 12
min_workers: 6
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -158,7 +158,7 @@ available_node_types:

tpu_slice_v5p_16:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-16
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -170,7 +170,7 @@ available_node_types:

tpu_slice_v5p_32:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-32
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -182,7 +182,7 @@ available_node_types:

tpu_slice_v5p_64:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-64
runtimeVersion: v2-alpha-tpuv5
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-us-central2-staging.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 4
min_workers: 2
node_config:
acceleratorType: v4-8
runtimeVersion: tpu-ubuntu2204-base
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-us-central2-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v4-8
runtimeVersion: tpu-ubuntu2204-base
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-us-central2.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 4
min_workers: 2
node_config:
acceleratorType: v4-8
runtimeVersion: tpu-ubuntu2204-base
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-us-east1-d-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v6e-8
runtimeVersion: v2-alpha-tpuv6e
Expand Down
4 changes: 2 additions & 2 deletions infra/marin-us-east5-a-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 1
min_workers: 0
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -141,7 +141,7 @@ available_node_types:

tpu_slice_v5p_8:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand Down
4 changes: 2 additions & 2 deletions infra/marin-us-east5-a.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 8
min_workers: 4
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand All @@ -146,7 +146,7 @@ available_node_types:

tpu_slice_v5p_8:
max_workers: 1024
min_workers: 8
min_workers: 4
node_config:
acceleratorType: v5p-8
runtimeVersion: v2-alpha-tpuv5
Expand Down
2 changes: 1 addition & 1 deletion infra/marin-us-east5-b-vllm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ available_node_types:
sourceImage: projects/ubuntu-os-cloud/global/images/family/ubuntu-2204-lts
tpu_worker:
max_workers: 1024
min_workers: 2
min_workers: 1
node_config:
acceleratorType: v6e-8
runtimeVersion: v2-alpha-tpuv6e
Expand Down
2 changes: 1 addition & 1 deletion lib/iris/examples/marin-dev.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ controller:
image: ghcr.io/marin-community/iris-controller:latest
gcp:
zone: us-central1-a
machine_type: e2-standard-4
machine_type: e2-highmem-4
boot_disk_size_gb: 100
port: 10000

Expand Down
14 changes: 7 additions & 7 deletions lib/iris/examples/marin.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ controller:
image: ghcr.io/marin-community/iris-controller:latest
gcp:
zone: us-central1-a
machine_type: e2-standard-4
machine_type: e2-highmem-4
boot_disk_size_gb: 100
port: 10000

Expand All @@ -42,7 +42,7 @@ scale_groups:
priority: 1000
resources: { cpu: 2, ram: 16GB, disk: 100GB, device_type: cpu, preemptible: false }
min_slices: 0
max_slices: 1
max_slices: 4
slice_template:
gcp:
mode: GCP_SLICE_MODE_VM
Expand All @@ -57,7 +57,7 @@ scale_groups:
num_vms: 1
priority: 10
resources: { cpu: 112, ram: 192GB, disk: 100GB, device_type: tpu, device_variant: v5litepod-4, device_count: 4, preemptible: true }
min_slices: 1
min_slices: 3
max_slices: 1024
slice_template:
gcp:
Expand Down Expand Up @@ -107,7 +107,7 @@ scale_groups:
num_vms: 32
priority: 60
resources: { cpu: 112, ram: 192GB, disk: 100GB, device_type: tpu, device_variant: v5litepod-128, device_count: 4, preemptible: true }
min_slices: 0
min_slices: 1
max_slices: 32
slice_template:
gcp:
Expand Down Expand Up @@ -182,7 +182,7 @@ scale_groups:
num_vms: 32
priority: 60
resources: { cpu: 180, ram: 720GB, disk: 100GB, device_type: tpu, device_variant: v6e-128, device_count: 4, preemptible: true }
min_slices: 0
min_slices: 1
max_slices: 32
slice_template:
gcp:
Expand All @@ -208,7 +208,7 @@ scale_groups:
num_vms: 1
priority: 20
resources: { cpu: 208, ram: 448GB, disk: 100GB, device_type: tpu, device_variant: v5p-8, device_count: 4, preemptible: true }
min_slices: 0
min_slices: 8
max_slices: 2048
slice_template:
gcp:
Expand Down Expand Up @@ -303,7 +303,7 @@ scale_groups:
num_vms: 1
priority: 20
resources: { cpu: 240, ram: 400GB, disk: 100GB, device_type: tpu, device_variant: v4-8, device_count: 4, preemptible: true }
min_slices: 0
min_slices: 2
max_slices: 2048
slice_template:
gcp:
Expand Down
Loading