mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
Add a new autoscaling shuffle test (#16948)
This commit is contained in:
parent
4bde71ca86
commit
fa0ff057d6
2 changed files with 29 additions and 0 deletions
|
@ -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:
|
||||
|
|
|
@ -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
|
Loading…
Add table
Reference in a new issue