xref: /aosp_15_r20/external/pytorch/.circleci/scripts/binary_windows_build.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1#!/bin/bash
2set -eux -o pipefail
3
4source "${BINARY_ENV_FILE:-/c/w/env}"
5mkdir -p "$PYTORCH_FINAL_PACKAGE_DIR"
6
7export CUDA_VERSION="${DESIRED_CUDA/cu/}"
8export USE_SCCACHE=1
9export SCCACHE_BUCKET=ossci-compiler-cache
10export SCCACHE_IGNORE_SERVER_IO_ERROR=1
11export VC_YEAR=2019
12
13if [[ "$DESIRED_CUDA" == 'xpu' ]]; then
14    export VC_YEAR=2022
15    export USE_SCCACHE=0
16fi
17
18echo "Free space on filesystem before build:"
19df -h
20
21pushd "$BUILDER_ROOT"
22if [[ "$PACKAGE_TYPE" == 'conda' ]]; then
23    ./windows/internal/build_conda.bat
24elif [[ "$PACKAGE_TYPE" == 'wheel' || "$PACKAGE_TYPE" == 'libtorch' ]]; then
25    export NIGHTLIES_PYTORCH_ROOT="$PYTORCH_ROOT"
26    ./windows/internal/build_wheels.bat
27fi
28
29echo "Free space on filesystem after build:"
30df -h
31