commit b7363c077739e5080ad7f0af04366cc4254cec38
author Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Wed Jan 22 02:23:18 2020 +0000
committer Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Wed Jan 22 02:23:18 2020 +0000
tree 9fd654213f59fc8cb0843095c514dde4e22e65c2
parent 559bc51f4668039c62daf6f1c80855ad5315e7ac
parent 55bcdf84c5d3b052c4ce1e5ca665bccf4f7a9c8f

Merge "Make host image build script check zone for GPUs" am: 531ed7e264 am: 668d9a7c03 am: 55bcdf84c5

Change-Id: Id4800fec678706af02df69e3c43d76c938fbb2b0

tools/create_base_image_hostlib.sh

diff --git a/tools/create_base_image_hostlib.sh b/tools/create_base_image_hostlib.sh
index d3930db..f9982a5 100755
--- a/tools/create_base_image_hostlib.sh
+++ b/tools/create_base_image_hostlib.sh
@@ -37,6 +37,11 @@
 
 SSH_FLAGS=(${INTERNAL_IP})
 
+fatal_echo() {
+  echo "$1"
+  exit 1
+}
+
 wait_for_instance() {
   alive=""
   while [[ -z "${alive}" ]]; do
@@ -96,13 +101,16 @@
     --image-family="${FLAGS_source_image_family}" \
     --image-project="${FLAGS_source_image_project}" \
     "${FLAGS_dest_image}"
+  local gpu_type="nvidia-tesla-p100-vws"
+  gcloud compute accelerator-types describe "${gpu_type}" "${PZ[@]}" || \
+    fatal_echo "Please use a zone with ${gpu_type} GPUs available."
   gcloud compute instances create \
     "${PZ[@]}" \
     --machine-type=n1-standard-16 \
     --image-family="${FLAGS_source_image_family}" \
     --image-project="${FLAGS_source_image_project}" \
     --boot-disk-size=200GiB \
-    --accelerator="type=nvidia-tesla-p100-vws,count=1" \
+    --accelerator="type=${gpu_type},count=1" \
     --maintenance-policy=TERMINATE \
     "${FLAGS_build_instance}"
   wait_for_instance "${PZ[@]}" "${FLAGS_build_instance}"
@@ -135,7 +143,7 @@
       --image="${FLAGS_dest_image}" \
       --machine-type=n1-standard-4 \
       --scopes storage-ro \
-      --accelerator="type=nvidia-tesla-p100-vws,count=1" \
+      --accelerator="type=${gpu_type},count=1" \
       --maintenance-policy=TERMINATE \
       "${FLAGS_launch_instance}"
   fi