xref: /aosp_15_r20/external/pytorch/scripts/buck_setup.sh (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1#!/bin/bash
2printf "\nCreating .buckconfig\n"
3cp .buckconfig.oss .buckconfig
4
5PROXY=""
6if [ "$1" == "devserver" ]; then
7   echo -e '\n[download]\n   proxy_host=fwdproxy\n   proxy_port=8080\n   proxy_type=HTTP\n' >> .buckconfig
8   PROXY="$(fwdproxy-config curl)"
9   printf "using proxy $PROXY\n\n"
10fi
11
12cat .buckconfig
13
14cd third_party || return
15
16printf "\nGenerating cpuinfo wrappers\n"
17python3 generate-cpuinfo-wrappers.py
18
19printf "\nGenerating xnnpack wrappers\n"
20python3 generate-xnnpack-wrappers.py
21
22# bazel-skylib
23printf "\nDownloading bazel-skylib\n"
24rm -rf bazel-skylib; mkdir bazel-skylib
25curl --retry 3 -L $PROXY https://github.com/bazelbuild/bazel-skylib/releases/download/1.0.2/bazel-skylib-1.0.2.tar.gz|tar zx -C bazel-skylib
26
27# glog
28printf "\nDownloading glog\n"
29rm -rf glog; mkdir glog
30curl --retry 3 -L $PROXY https://github.com/google/glog/archive/v0.4.0.tar.gz | tar zx -C glog --strip-components 1
31
32# ruy
33printf "\nDownloading ruy\n"
34curl --retry 3 -L $PROXY -o /tmp/ruy.zip https://github.com/google/ruy/archive/a09683b8da7164b9c5704f88aef2dc65aa583e5d.zip
35unzip -q /tmp/ruy.zip -d /tmp/
36rm -rf ruy/
37mv /tmp/ruy-a09683b8da7164b9c5704f88aef2dc65aa583e5d ruy/
38