Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ci][tune][train] Update release test compute configs to not schedule work on head node #48103

Merged
8 changes: 5 additions & 3 deletions release/air_tests/air_benchmarks/compute_cpu_4_aws.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,17 @@
cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
region: us-west-2

max_workers: 3
max_workers: 4

head_node_type:
name: head_node
instance_type: m5.2xlarge
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: m5.2xlarge
max_workers: 3
min_workers: 3
max_workers: 4
min_workers: 4
use_spot: false
8 changes: 5 additions & 3 deletions release/air_tests/air_benchmarks/compute_cpu_4_gce.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,17 @@ region: us-west1
allowed_azs:
- us-west1-b

max_workers: 3
max_workers: 4

head_node_type:
name: head_node
instance_type: n1-standard-8
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: n1-standard-8
max_workers: 3
min_workers: 3
max_workers: 4
min_workers: 4
use_spot: false
6 changes: 4 additions & 2 deletions release/air_tests/air_benchmarks/compute_cpu_8_aws.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,12 @@ max_workers: 7
head_node_type:
name: head_node
instance_type: m5.2xlarge
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: m5.2xlarge
max_workers: 7
min_workers: 7
max_workers: 8
min_workers: 8
use_spot: false
8 changes: 5 additions & 3 deletions release/air_tests/air_benchmarks/compute_cpu_8_gce.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,17 @@ region: us-west1
allowed_azs:
- us-west1-b

max_workers: 7
max_workers: 8

head_node_type:
name: head_node
instance_type: n1-standard-8
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: n1-standard-8
max_workers: 7
min_workers: 7
max_workers: 8
min_workers: 8
use_spot: false
10 changes: 6 additions & 4 deletions release/air_tests/air_benchmarks/compute_gpu_2x2_aws.yaml
Original file line number Diff line number Diff line change
@@ -1,15 +1,17 @@
cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
region: us-west-2

max_workers: 1
max_workers: 2

head_node_type:
name: head_node
instance_type: g3.8xlarge
instance_type: m5.2xlarge
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: g3.8xlarge
max_workers: 1
min_workers: 1
max_workers: 2
min_workers: 2
use_spot: false
10 changes: 6 additions & 4 deletions release/air_tests/air_benchmarks/compute_gpu_2x2_gce.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,15 +3,17 @@ region: us-west1
allowed_azs:
- us-west1-b

max_workers: 1
max_workers: 2

head_node_type:
name: head_node
instance_type: n1-standard-32-nvidia-tesla-t4-2
instance_type: n1-standard-8
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: n1-standard-32-nvidia-tesla-t4-2
max_workers: 1
min_workers: 1
max_workers: 2
min_workers: 2
use_spot: false
10 changes: 6 additions & 4 deletions release/air_tests/air_benchmarks/compute_gpu_4x4_aws.yaml
Original file line number Diff line number Diff line change
@@ -1,17 +1,19 @@
cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
region: us-west-2

max_workers: 3
max_workers: 4

head_node_type:
name: head_node
instance_type: g4dn.12xlarge
instance_type: m5.2xlarge
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: g4dn.12xlarge
max_workers: 3
min_workers: 3
max_workers: 4
min_workers: 4
use_spot: false

aws:
Expand Down
10 changes: 6 additions & 4 deletions release/air_tests/air_benchmarks/compute_gpu_4x4_gce.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,19 @@ region: us-west1
allowed_azs:
- us-west1-b

max_workers: 3
max_workers: 4

head_node_type:
name: head_node
instance_type: n1-standard-64-nvidia-tesla-t4-4
instance_type: n1-standard-8
resources:
cpu: 0

worker_node_types:
- name: worker_node
instance_type: n1-standard-64-nvidia-tesla-t4-4
max_workers: 3
min_workers: 3
max_workers: 4
min_workers: 4
use_spot: false

gcp_advanced_configurations_json:
Expand Down