saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add sccache support for hcc and hip-clang in ROCm (#38451)

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/38451

Differential Revision: D21589126

Pulled By: ezyang

fbshipit-source-id: dc4d08e7f393dbe369e501334c776071b2c176e0
This commit is contained in:
Pruthvi Madugundu 2020-05-14 20:41:56 -07:00 committed by Facebook GitHub Bot
parent fac9f36563
commit 8df14c573e
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
docker/caffe2/jenkins/common/install_ccache.sh
View file

@ -14,17 +14,20 @@ if [ -z "${SCCACHE}" ]; then
exit 1
fi
# If rocm build, add hcc to sccache.
# If rocm build, if hcc file exists then use hcc else clang(hip-clang) for sccache
if [[ "${BUILD_ENVIRONMENT}" == *-rocm* ]]; then
# HCC's symlink path: /opt/rocm/hcc/bin/hcc -> /opt/rocm/hcc/bin/clang -> /opt/rocm/hcc/bin/clang-7.0
HCC_DEST_PATH="$(readlink -f $(which hcc))"
HCC_REAL_BINARY="$(dirname $HCC_DEST_PATH)/clang-7.0_original"
mv "$HCC_DEST_PATH" "$HCC_REAL_BINARY"
if [[ -e "/opt/rocm/hcc/bin/hcc" ]]; then
HIPCOM_DEST_PATH="$(readlink -f /opt/rocm/hcc/bin/hcc )"
else
HIPCOM_DEST_PATH="$(readlink -f /opt/rocm/llvm/bin/clang )"
fi
HIPCOM_REAL_BINARY="$(dirname $HIPCOM_DEST_PATH)/hipcompiler_original"
mv "$HIPCOM_DEST_PATH" "$HIPCOM_REAL_BINARY"
# Create sccache wrapper.
(
echo "#!/bin/sh"
echo "exec $SCCACHE $HCC_REAL_BINARY \"\$@\""
) > "$HCC_DEST_PATH"
chmod +x "$HCC_DEST_PATH"
echo "exec $SCCACHE $HIPCOM_REAL_BINARY \"\$@\""
) > "$HIPCOM_DEST_PATH"
chmod +x "$HIPCOM_DEST_PATH"
fi