Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Correct JVM CMake GPU flag. (#4071)

Browse Source
This commit is contained in:
Jiaming Yuan 2019-01-21 20:36:38 +08:00 committed by GitHub
parent 1fc37e4749
commit 301cef4638
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
CMakeLists.txt
View File

@ -8,17 +8,23 @@ set_default_configuration_release()
msvc_use_static_runtime() msvc_use_static_runtime()
# Options # Options
option(USE_CUDA "Build with GPU acceleration") ## GPUs
option(JVM_BINDINGS "Build JVM bindings" OFF) option(USE_CUDA "Build with GPU acceleration" OFF)
option(GOOGLE_TEST "Build google tests" OFF) option(USE_NCCL "Build with multiple GPUs support" OFF)
option(R_LIB "Build shared library for R package" OFF)
set(GPU_COMPUTE_VER "" CACHE STRING set(GPU_COMPUTE_VER "" CACHE STRING
"Space separated list of compute versions to be built against, e.g. '35 61'") "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(USE_SANITIZER "Use santizer flags" OFF)
option(SANITIZER_PATH "Path to sanitizes.") option(SANITIZER_PATH "Path to sanitizes.")
set(ENABLED_SANITIZERS "address" "leak" CACHE STRING set(ENABLED_SANITIZERS "address" "leak" CACHE STRING
"Semicolon separated list of sanitizer names. E.g 'address;leak'. Supported sanitizers are "Semicolon separated list of sanitizer names. E.g 'address;leak'. Supported sanitizers are
address, leak and thread.") address, leak and thread.")
option(GOOGLE_TEST "Build google tests" OFF)
# Plugins # Plugins
option(PLUGIN_LZ4 "Build lz4 plugin" OFF) option(PLUGIN_LZ4 "Build lz4 plugin" OFF)

2
jvm-packages/create_jni.py
View File

@ -19,7 +19,7 @@ CONFIG = {
"USE_AZURE": "OFF", "USE_AZURE": "OFF",
"USE_S3": "OFF", "USE_S3": "OFF",
"PLUGIN_UPDATER_GPU": "OFF", "USE_CUDA": "OFF",
"JVM_BINDINGS": "ON" "JVM_BINDINGS": "ON"
} }