[BUILD] Include rocm cross compile env (#367)

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"

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
 

tests/ci_build/install/ubuntu_install_rocm.sh

+# 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