Merge pull request #43726 from vishh/local-ssd-gce

cluster/gce/config-default.sh

@@ -31,6 +31,7 @@ MASTER_DISK_TYPE=pd-ssd
 MASTER_DISK_SIZE=${MASTER_DISK_SIZE:-20GB}
 NODE_DISK_TYPE=${NODE_DISK_TYPE:-pd-standard}
 NODE_DISK_SIZE=${NODE_DISK_SIZE:-100GB}
+NODE_LOCAL_SSDS=${NODE_LOCAL_SSDS:-0}
 REGISTER_MASTER_KUBELET=${REGISTER_MASTER:-true}
 PREEMPTIBLE_NODE=${PREEMPTIBLE_NODE:-false}
 PREEMPTIBLE_MASTER=${PREEMPTIBLE_MASTER:-false}

cluster/gce/config-test.sh

@@ -31,6 +31,7 @@ MASTER_DISK_TYPE=pd-ssd
 MASTER_DISK_SIZE=${MASTER_DISK_SIZE:-20GB}
 NODE_DISK_TYPE=${NODE_DISK_TYPE:-pd-standard}
 NODE_DISK_SIZE=${NODE_DISK_SIZE:-100GB}
+NODE_LOCAL_SSDS=${NODE_LOCAL_SSDS:-0}
 REGISTER_MASTER_KUBELET=${REGISTER_MASTER:-true}
 KUBE_APISERVER_REQUEST_TIMEOUT=300
 PREEMPTIBLE_NODE=${PREEMPTIBLE_NODE:-false}

cluster/gce/util.sh

@@ -480,6 +480,12 @@ function create-node-template() {
   if [[ "${PREEMPTIBLE_NODE}" == "true" ]]; then
     preemptible_minions="--preemptible --maintenance-policy TERMINATE"
   fi
+  local local_ssds=""
+  if [ ! -z ${NODE_LOCAL_SSDS+x} ]; then
+      for i in $(seq ${NODE_LOCAL_SSDS}); do
+          local_ssds="$local_ssds--local-ssd=interface=SCSI "
+      done
+  fi
   while true; do
     echo "Attempt ${attempt} to create ${1}" >&2
     if ! gcloud compute instance-templates create "$template_name" \
@@ -491,6 +497,7 @@ function create-node-template() {
       --image "${NODE_IMAGE}" \
       --tags "${NODE_TAG}" \
       --network "${NETWORK}" \
+      ${local_ssds} \
       ${preemptible_minions} \
       $2 \
       --can-ip-forward \