From 1957e6fb4dd823f94251e7378abef03cba05f545 Mon Sep 17 00:00:00 2001 From: Rory Mitchell Date: Fri, 23 Dec 2016 21:18:56 +1300 Subject: [PATCH] Fix cmake build for linux. Update GPU benchmarks. (#1904) --- CMakeLists.txt | 2 +- plugin/updater_gpu/README.md | 9 +-------- 2 files changed, 2 insertions(+), 9 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 355b060b7..f1318ddc6 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -5,7 +5,7 @@ find_package(OpenMP) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}") if(NOT MSVC) - set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC") + set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -O3 -funroll-loops -msse2") endif() diff --git a/plugin/updater_gpu/README.md b/plugin/updater_gpu/README.md index 9a31869aa..1bc7d982c 100644 --- a/plugin/updater_gpu/README.md +++ b/plugin/updater_gpu/README.md @@ -2,14 +2,7 @@ ## Benchmarks -Time for 500 boosting iterations in seconds. - -Dataset | Instances | Features | i7-6700K | Titan X (pascal) | Speedup ---- | --- | --- | --- | --- | --- -Yahoo LTR | 473,134 | 700 | 3738 | 507 | 7.37 -Higgs | 10,500,000 | 28 | 31352 | 4173 | 7.51 -Bosch | 1,183,747 | 968 | 9460 | 1009 | 9.38 - +[See here](http://dmlc.ml/2016/12/14/GPU-accelerated-xgboost.html) for performance benchmarks ## Usage Specify the updater parameter as 'grow_gpu'.