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

Updates for GPU CI tests (#3467)

Browse Source 
* Fail GPU CI after test failure

* Fix GPU linear tests

* Reduced number of GPU tests to speed up CI

* Remove static allocations of device memory

* Resolve illegal memory access for updater_fast_hist.cc

* Fix broken r tests dependency

* Update python install documentation for GPU
This commit is contained in:
Rory Mitchell
2018-07-16 18:05:53 +12:00
committed by GitHub
parent a13e29ece1
commit 1b59316444
7 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/python-gpu/test_gpu_linear.py
View File

@@ -9,6 +9,10 @@ import unittest
class TestGPULinear(unittest.TestCase):
def test_gpu_coordinate(self):
tm._skip_if_no_sklearn()
variable_param = {'alpha': [.005, .1], 'lambda': [0.005],
'coordinate_selection': ['cyclic', 'random', 'greedy'], 'n_gpus': [-1, 1]}
test_linear.assert_updater_accuracy('gpu_coord_descent', variable_param)
variable_param = {'booster': ['gblinear'], 'updater': ['coord_descent'], 'eta': [0.5],
'top_k': [10], 'tolerance': [1e-5], 'nthread': [2], 'alpha': [.005, .1], 'lambda': [0.005],
'coordinate_selection': ['cyclic', 'random', 'greedy'], 'n_gpus': [-1]}
for param in test_linear.parameter_combinations(variable_param):
results = test_linear.run_suite(param, 200, None, scale_features=True)
test_linear.assert_regression_result(results, 1e-2)
test_linear.assert_classification_result(results)