From 301cef463826d24837d4f17c742b6c9e4502dbe4 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Mon, 21 Jan 2019 20:36:38 +0800
Subject: [PATCH] Correct JVM CMake GPU flag. (#4071)

---
 CMakeLists.txt             | 14 ++++++++++----
 jvm-packages/create_jni.py |  2 +-
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 061431de5..aec9c0664 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -8,17 +8,23 @@ set_default_configuration_release()
 msvc_use_static_runtime()
 
 # Options
-option(USE_CUDA  "Build with GPU acceleration")
-option(JVM_BINDINGS "Build JVM bindings" OFF)
-option(GOOGLE_TEST "Build google tests" OFF)
-option(R_LIB "Build shared library for R package" OFF)
+## GPUs
+option(USE_CUDA  "Build with GPU acceleration" OFF)
+option(USE_NCCL  "Build with multiple GPUs support" OFF)
 set(GPU_COMPUTE_VER "" CACHE STRING
   "Space separated list of compute versions to be built against, e.g. '35 61'")
+
+## Bindings
+option(JVM_BINDINGS "Build JVM bindings" OFF)
+option(R_LIB "Build shared library for R package" OFF)
+
+## Devs
 option(USE_SANITIZER "Use santizer flags" OFF)
 option(SANITIZER_PATH "Path to sanitizes.")
 set(ENABLED_SANITIZERS "address" "leak" CACHE STRING
   "Semicolon separated list of sanitizer names. E.g 'address;leak'. Supported sanitizers are
 address, leak and thread.")
+option(GOOGLE_TEST "Build google tests" OFF)
 
 # Plugins
 option(PLUGIN_LZ4 "Build lz4 plugin" OFF)
diff --git a/jvm-packages/create_jni.py b/jvm-packages/create_jni.py
index 8ce0aa1bc..b16e68e1e 100755
--- a/jvm-packages/create_jni.py
+++ b/jvm-packages/create_jni.py
@@ -19,7 +19,7 @@ CONFIG = {
     "USE_AZURE": "OFF",
     "USE_S3": "OFF",
 
-    "PLUGIN_UPDATER_GPU": "OFF",
+    "USE_CUDA": "OFF",
     "JVM_BINDINGS": "ON"
 }