| #!/bin/bash |
| |
| # Copyright 2022 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| # Updates the autoscaling configuration for a specific instance group. |
| |
| function update_autoscaling() { |
| local mig_name="$1" |
| local region="$2" |
| local min_size="$3" |
| local max_size="$4" |
| local -a autoscaling_args=( |
| "${mig_name}" |
| --project=iree-oss |
| --region="${region}" |
| --cool-down-period=60 |
| --min-num-replicas="${min_size}" |
| --max-num-replicas="${max_size}" |
| --mode=only-scale-out |
| --target-cpu-utilization=0.2 |
| ) |
| |
| (set -x; gcloud beta compute instance-groups managed set-autoscaling "${autoscaling_args[@]}") |
| } |
| |
| update_autoscaling "$@" |
