From 90e22393722318700c1406650fe38468f87cdcb8 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Tue, 5 Nov 2019 01:27:08 -0500
Subject: [PATCH] Fix cmake variable. (#126)

---
 CMakeLists.txt | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d668be3fb..541b990ed 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2,6 +2,11 @@ cmake_minimum_required(VERSION 3.3)
 
 project(rabit VERSION 0.3.0 LANGUAGES CXX)
 
+if ((${CMAKE_VERSION} VERSION_GREATER 3.13) OR (${CMAKE_VERSION} VERSION_EQUAL 3.13))
+  # This allows user to specify `RABIT_BUILD_DMLC` and others as CMake variable.
+  cmake_policy(SET CMP0077 NEW)
+endif ((${CMAKE_VERSION} VERSION_GREATER 3.13) OR (${CMAKE_VERSION} VERSION_EQUAL 3.13))
+
 option(RABIT_BUILD_TESTS "Build rabit tests" OFF)
 option(RABIT_BUILD_MPI "Build MPI" OFF)
 option(RABIT_BUILD_DMLC "Include DMLC_CORE in build" OFF)
@@ -73,7 +78,7 @@ endif(DMLC_ROOT)
 
 foreach(lib ${rabit_libs})
   target_include_directories(${lib} PUBLIC
-          "$<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}/include>"
+          "$<BUILD_INTERFACE:${rabit_SOURCE_DIR}/include>"
           "$<BUILD_INTERFACE:${DMLC_ROOT}/include>")
 endforeach()