From a2fea331035dacba13cdb6889418f5ef193e77a0 Mon Sep 17 00:00:00 2001
From: odidev <odidev@puresoftware.com>
Date: Wed, 7 Oct 2020 12:32:09 +0530
Subject: [PATCH] Added arm64 job in Travis-CI (#6200)

Signed-off-by: odidev <odidev@puresoftware.com>
---
 .travis.yml              | 3 +++
 tests/travis/run_test.sh | 4 ++++
 tests/travis/setup.sh    | 4 +++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/.travis.yml b/.travis.yml
index 49f23dd98..65a572e1d 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -12,6 +12,9 @@ jobs:
     - os: linux
       arch: amd64
       env: TASK=python_sdist_test
+    - os: linux
+      arch: arm64
+      env: TASK=python_sdist_test
     - os: osx
       arch: amd64
       osx_image: xcode10.1
diff --git a/tests/travis/run_test.sh b/tests/travis/run_test.sh
index 1db1c73b7..f65eeff50 100755
--- a/tests/travis/run_test.sh
+++ b/tests/travis/run_test.sh
@@ -10,6 +10,10 @@ if [ ${TASK} == "python_sdist_test" ]; then
     conda activate python3
     python --version
     conda install numpy scipy
+    if [ ${TRAVIS_CPU_ARCH} == "arm64" ]; then
+        conda install cmake
+        cmake --version
+    fi
 
     make pippack
     python -m pip install xgboost-*.tar.gz -v --user
diff --git a/tests/travis/setup.sh b/tests/travis/setup.sh
index 2056bc78f..62f1feff5 100755
--- a/tests/travis/setup.sh
+++ b/tests/travis/setup.sh
@@ -3,6 +3,8 @@
 if [ ${TASK} == "python_test" ] || [ ${TASK} == "python_sdist_test" ]; then
     if [ ${TRAVIS_OS_NAME} == "osx" ]; then
         wget -O conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh
+    elif [ ${TRAVIS_CPU_ARCH} == "arm64" ]; then
+        wget -O conda.sh https://github.com/conda-forge/miniforge/releases/download/4.8.2-1/Miniforge3-4.8.2-1-Linux-aarch64.sh
     else
         wget -O conda.sh https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh
     fi
@@ -25,7 +27,7 @@ if [ ${TASK} == "s390x_test" ] && [ ${TRAVIS_CPU_ARCH} == "s390x" ]; then
       time python3 python3-pip python3-numpy python3-scipy python3-sklearn r-base
 fi
 
-if [ ${TASK} == "python_sdist_test" ] && [ ${TRAVIS_OS_NAME} == "linux" ]; then
+if [ ${TASK} == "python_sdist_test" ] && [ ${TRAVIS_OS_NAME} == "linux" ] && [ ${TRAVIS_CPU_ARCH} == "amd64" ]; then
     wget https://github.com/Kitware/CMake/releases/download/v3.17.1/cmake-3.17.1-Linux-x86_64.sh
     sudo bash cmake-3.17.1-Linux-x86_64.sh --prefix=/usr/local --skip-license
     sudo rm -rf /usr/local/cmake-3.12.4  # Remove existing CMake