From 1772e072721f7a931745bca8adb3ccb29e2b4b0b Mon Sep 17 00:00:00 2001 From: Jorghi12 Date: Tue, 21 Aug 2018 16:31:43 -0400 Subject: [PATCH] Enable LAPACK --- .jenkins/pytorch/build.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.jenkins/pytorch/build.sh b/.jenkins/pytorch/build.sh index bfbd40b259fbc3..7fec3105de0e04 100755 --- a/.jenkins/pytorch/build.sh +++ b/.jenkins/pytorch/build.sh @@ -48,6 +48,10 @@ if [[ "$BUILD_ENVIRONMENT" == *rocm* ]]; then sudo apt-get install libc++abi1 python tools/amd_build/build_pytorch_amd.py + + # Enable LAPACK + pip install mkl mkl-devel + USE_ROCM=1 python setup.py install --user exit 0 fi