From 2a87020cce14d405653e373fb55ba748426bd7ca Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Mon, 21 Aug 2017 20:39:09 -0700
Subject: [PATCH] [BUILD] Include rocm cross compile env (#367)
---
Jenkinsfile | 2 ++
tests/ci_build/Dockerfile.gpu | 3 +++
tests/ci_build/install/ubuntu_install_rocm.sh | 4 ++++
3 files changed, 9 insertions(+)
create mode 100644 tests/ci_build/install/ubuntu_install_rocm.sh
diff --git a/Jenkinsfile b/Jenkinsfile
index daa810bbd..039ef1d4a 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -91,6 +91,8 @@ stage('Build') {
echo LLVM_CONFIG=llvm-config-4.0 >> config.mk
echo USE_RPC=1 >> config.mk
echo USE_BLAS=openblas >> config.mk
+ echo USE_ROCM=1 >> config.mk
+ echo ROCM_PATH=/opt/rocm >> config.mk
"""
make('gpu', '-j2')
sh "mv lib/libtvm.so lib/libtvm_llvm40.so"
diff --git a/tests/ci_build/Dockerfile.gpu b/tests/ci_build/Dockerfile.gpu
index 7e3377b7d..19e18abb0 100644
--- a/tests/ci_build/Dockerfile.gpu
+++ b/tests/ci_build/Dockerfile.gpu
@@ -34,6 +34,9 @@ RUN bash /install/ubuntu_install_java.sh
COPY install/ubuntu_install_nodejs.sh /install/ubuntu_install_nodejs.sh
RUN bash /install/ubuntu_install_nodejs.sh
+COPY install/ubuntu_install_rocm.sh /install/ubuntu_install_rocm.sh
+RUN bash /install/ubuntu_install_rocm.sh
+
# Enable doxygen for c++ doc build
RUN apt-get install -y doxygen graphviz
diff --git a/tests/ci_build/install/ubuntu_install_rocm.sh b/tests/ci_build/install/ubuntu_install_rocm.sh
new file mode 100644
index 000000000..d050c2007
--- /dev/null
+++ b/tests/ci_build/install/ubuntu_install_rocm.sh
@@ -0,0 +1,4 @@
+# Install ROCm cross compilation toolchain.
+wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
+echo deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main > /etc/apt/sources.list.d/rocm.list
+apt-get update && apt-get install -y --force-yes rocm-dev
--
GitLab