diff options
| author | Junjie Bai <bai@in.tum.de> | 2018-08-22 12:54:33 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-08-22 13:15:45 -0700 |
| commit | 754ec9e386f9a168edcd6d8eb4a25d8d5181da10 (patch) | |
| tree | c36e26f84a38a064c8a53740d1087e1b9716e3cd /.jenkins | |
| parent | 9767951ca8a0af26d3caf2e84a52b10f81b9c2cb (diff) | |
| download | pytorch-754ec9e386f9a168edcd6d8eb4a25d8d5181da10.tar.gz pytorch-754ec9e386f9a168edcd6d8eb4a25d8d5181da10.tar.bz2 pytorch-754ec9e386f9a168edcd6d8eb4a25d8d5181da10.zip | |
Reduce rocm link time with ThinLTO
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10758
Differential Revision: D9467554
Pulled By: bddppq
fbshipit-source-id: 6853ccd96ac3209e062c110913ea37d6840c8134
Diffstat (limited to '.jenkins')
| -rwxr-xr-x | .jenkins/caffe2/build.sh | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/.jenkins/caffe2/build.sh b/.jenkins/caffe2/build.sh index 928c5fa0e4..5f5faa3759 100755 --- a/.jenkins/caffe2/build.sh +++ b/.jenkins/caffe2/build.sh @@ -156,6 +156,11 @@ if [[ $BUILD_ENVIRONMENT == *rocm* ]]; then export LC_ALL=C.UTF-8 export HCC_AMDGPU_TARGET=gfx900 + # The link time of libcaffe2_hip.so takes 40 minutes, according to + # https://github.com/RadeonOpenCompute/hcc#thinlto-phase-1---implemented + # using using ThinLTO could significantly improve link-time performance. + export KMTHINLTO=1 + ########## HIPIFY Caffe2 operators ${PYTHON} "${ROOT_DIR}/tools/amd_build/build_pytorch_amd.py" ${PYTHON} "${ROOT_DIR}/tools/amd_build/build_caffe2_amd.py" |