testing/gpu_test_arm64.yaml

substitutions:
  '_TEST': 'gpu_test'
  '_INPUT_IMAGE': 'cos-arm64-lm-125-19216-104-23'
  '_INPUT_PROJECT': 'cos-cloud'
  '_ZONE': 'us-central1-a'
  '_PROJECT_NAME': ''
  '_SERVICE_ACCOUNT': 'compute@cloud-image-docker-builder.google.com.iam.gserviceaccount.com'
  '_COS_CUSTOMIZER': 'cos_customizer'
steps:
- name: 'gcr.io/cloud-builders/docker'
  entrypoint: '/bin/bash'
  args:
  - '-c'
  - |
    if [[ ${_COS_CUSTOMIZER} == 'cos_customizer' ]]; then
      docker build -f src/cmd/cos_customizer/Dockerfile -t cos_customizer .
    fi
- name: 'gcr.io/cloud-builders/gsutil'
  args: ['cp', 'gs://cos-nvidia-gpu-drivers/NVIDIA-Linux-aarch64-580.95.05.run', '.']
- name: '${_COS_CUSTOMIZER}'
  args: ['start-image-build',
         '-build-context=testing/${_TEST}',
         '-image-name=${_INPUT_IMAGE}',
         '-image-project=${_INPUT_PROJECT}',
         '-gcs-bucket=${_PROJECT_NAME}_cloudbuild',
         '-gcs-workdir=customizer-$BUILD_ID']
- name: '${_COS_CUSTOMIZER}'
  args: ['install-gpu',
         '-version=NVIDIA-Linux-aarch64-580.95.05.run',
         '-gpu-type=']
- name: '${_COS_CUSTOMIZER}'
  args: ['finish-image-build',
         '-zone=${_ZONE}',
         '-project=$PROJECT_ID',
         '-machine-type=t2a-standard-1',
         '-image-name=preload-test-$BUILD_ID',
         '-image-project=$PROJECT_ID',
         '-service-account=${_SERVICE_ACCOUNT}']
- name: 'gcr.io/cloud-builders/gcloud'
  args: ['compute', 'images', 'delete', 'preload-test-$BUILD_ID', '--project=$PROJECT_ID']
options:
  machineType: 'N1_HIGHCPU_32'
timeout: '7200s'