Update testnet-deploy script to configure GPUs for leader node (#1379)

ci/testnet-deploy.sh

@@ -4,6 +4,7 @@ cd "$(dirname "$0")"/..
 
 zone=
 leaderAddress=
+leaderMachineType=
 clientNodeCount=0
 validatorNodeCount=10
 publicNetwork=false
@@ -31,6 +32,7 @@ Deploys a CD testnet
    -n [number]          - Number of validator nodes (default: $validatorNodeCount)
    -c [number]          - Number of client nodes (default: $clientNodeCount)
    -P                   - Use public network IP addresses (default: $publicNetwork)
+   -G                   - Enable GPU, and set count/type of GPUs to use (e.g n1-standard-16 --accelerator count=4,type=nvidia-tesla-k80)
    -g                   - Enable GPU (default: $enableGpu)
    -a [address]         - Set the leader node's external IP address to this GCE address
    -d                   - Delete the network
@@ -47,7 +49,7 @@ zone=$2
 [[ -n $zone ]] || usage "Zone not specified"
 shift 2
 
-while getopts "h?p:Pn:c:s:ga:d" opt; do
+while getopts "h?p:Pn:c:s:gG:a:d" opt; do
   case $opt in
   h | \?)
     usage
@@ -74,6 +76,10 @@ while getopts "h?p:Pn:c:s:ga:d" opt; do
   g)
     enableGpu=true
     ;;
+  G)
+    enableGpu=true
+    leaderMachineType=$OPTARG
+    ;;
   a)
     leaderAddress=$OPTARG
     ;;
@@ -96,7 +102,11 @@ gce_create_args=(
 )
 
 if $enableGpu; then
-  gce_create_args+=(-g)
+  if [[ -z $leaderMachineType ]]; then
+    gce_create_args+=(-g)
+  else
+    gce_create_args+=(-G "$leaderMachineType")
+  fi
 fi
 
 if $publicNetwork; then