xref: /aosp_15_r20/external/pytorch/.ci/docker/common/install_vulkan_sdk.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1#!/bin/bash
2
3set -ex
4
5[ -n "${VULKAN_SDK_VERSION}" ]
6
7retry () {
8    $*  || (sleep 1 && $*) || (sleep 2 && $*) || (sleep 4 && $*) || (sleep 8 && $*)
9}
10
11_vulkansdk_dir=/var/lib/jenkins/vulkansdk
12_tmp_vulkansdk_targz=/tmp/vulkansdk.tar.gz
13
14curl \
15  --silent \
16  --show-error \
17  --location \
18  --fail \
19  --retry 3 \
20  --output "${_tmp_vulkansdk_targz}" "https://ossci-android.s3.amazonaws.com/vulkansdk-linux-x86_64-${VULKAN_SDK_VERSION}.tar.gz"
21
22mkdir -p "${_vulkansdk_dir}"
23tar -C "${_vulkansdk_dir}" -xzf "${_tmp_vulkansdk_targz}" --strip-components 1
24rm -rf "${_tmp_vulkansdk_targz}"
25