xref: /aosp_15_r20/external/pytorch/.ci/docker/common/install_swiftshader.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1#!/bin/bash
2
3set -ex
4
5[ -n "${SWIFTSHADER}" ]
6
7retry () {
8    $*  || (sleep 1 && $*) || (sleep 2 && $*) || (sleep 4 && $*) || (sleep 8 && $*)
9}
10
11_https_amazon_aws=https://ossci-android.s3.amazonaws.com
12
13# SwiftShader
14_swiftshader_dir=/var/lib/jenkins/swiftshader
15_swiftshader_file_targz=swiftshader-abe07b943-prebuilt.tar.gz
16mkdir -p $_swiftshader_dir
17_tmp_swiftshader_targz="/tmp/${_swiftshader_file_targz}"
18
19curl --silent --show-error --location --fail --retry 3 \
20  --output "${_tmp_swiftshader_targz}" "$_https_amazon_aws/${_swiftshader_file_targz}"
21
22tar -C "${_swiftshader_dir}" -xzf "${_tmp_swiftshader_targz}"
23
24export VK_ICD_FILENAMES="${_swiftshader_dir}/build/Linux/vk_swiftshader_icd.json"
25