Permalink
Browse files

Fix cmake build for linux. Update GPU benchmarks. (#1904)

  • Loading branch information...
1 parent fb0fc0c commit 1957e6fb4dd823f94251e7378abef03cba05f545 @RAMitchell RAMitchell committed with tqchen Dec 23, 2016
Showing with 2 additions and 9 deletions.
  1. +1 −1 CMakeLists.txt
  2. +1 −8 plugin/updater_gpu/README.md
View
@@ -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()
@@ -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'.

0 comments on commit 1957e6f

Please sign in to comment.