Add a new autoscaling shuffle test (#16948)

This commit is contained in:
Eric Liang 2021-07-12 16:54:38 -07:00 committed by GitHub
parent 4bde71ca86
commit fa0ff057d6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 0 deletions

View file

@ -99,6 +99,16 @@
prepare: python wait_cluster.py 4 600
script: python shuffle/shuffle_test.py --num-partitions=200 --partition-size=500e6 --no-streaming
# Test autoscaling 1TB streaming shuffle with a large number of partitions.
- name: autoscaling_shuffle_1tb_1000_partitions
cluster:
app_config: shuffle/shuffle_app_config.yaml
compute_template: shuffle/shuffle_compute_autoscaling.yaml
run:
timeout: 3000
script: python shuffle/shuffle_test.py --num-partitions=1000 --partition-size=1e9 --no-streaming
# Test multi nodes 1TB streaming shuffle with a large number of partitions.
- name: shuffle_1tb_1000_partition
cluster:

View file

@ -0,0 +1,19 @@
cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
region: us-west-2
aws:
BlockDeviceMappings:
- DeviceName: /dev/sda1
Ebs:
VolumeSize: 500
head_node_type:
name: head_node
instance_type: i3.4xlarge
worker_node_types:
- name: worker_node
instance_type: i3.4xlarge
min_workers: 0
max_workers: 19
use_spot: false