Skip to content

Commit 9eb3991

Browse files
authored
Merge pull request #42 from Jorghi12/fix_oom
Alleviating the LLVM Out of Memory Error.
2 parents dc3779f + 3d37465 commit 9eb3991

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.jenkins/pytorch/build.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,10 @@ if [[ "$BUILD_ENVIRONMENT" == *rocm* ]]; then
4646
sudo chown -R jenkins:jenkins /usr/local
4747
rm -rf "$(dirname "${BASH_SOURCE[0]}")/../../../pytorch_amd/" || true
4848
python "$(dirname "${BASH_SOURCE[0]}")/../../tools/amd_build/build_pytorch_amd.py"
49+
50+
# ROCm builds experience OOM issues when buliding with sscache. (HCC Issue #785)
51+
export MAX_JOBS=`expr $(nproc) - 1`
52+
4953
USE_ROCM=1 python setup.py install
5054
exit
5155
fi

0 commit comments

Comments
 (0)