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

6
tests/python-gpu/test_gpu_updaters.py
View File

@@ -29,9 +29,9 @@ class TestGPU(unittest.TestCase):
assert_gpu_results(cpu_results, gpu_results)
def test_gpu_hist(self):
variable_param = {'n_gpus': [1, -1], 'max_depth': [2, 6], 'max_leaves': [255, 4],
'max_bin': [2, 16, 1024],
'grow_policy': ['depthwise', 'lossguide']}
variable_param = {'n_gpus': [-1], 'max_depth': [2, 10], 'max_leaves': [255, 4],
'max_bin': [2, 256],
'grow_policy': ['lossguide']}
for param in parameter_combinations(variable_param):
param['tree_method'] = 'gpu_hist'
gpu_results = run_suite(param, select_datasets=datasets)