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

[CI] Reduce testing load with RMM (#6249)

Browse Source 
* [CI] Reduce testing load with RMM

* Address reviewer's comment
This commit is contained in:
Philip Hyunsu Cho 2020-10-18 19:16:46 -07:00 committed by GitHub
parent 549f361b71
commit 65ea42bd42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/python-gpu/conftest.py
View File

@ -41,3 +41,15 @@ def local_cuda_cluster(request, pytestconfig):
def pytest_addoption(parser): def pytest_addoption(parser):
parser.addoption('--use-rmm-pool', action='store_true', default=False, help='Use RMM pool') parser.addoption('--use-rmm-pool', action='store_true', default=False, help='Use RMM pool')
def pytest_collection_modifyitems(config, items):
if config.getoption('--use-rmm-pool'):
blocklist = [
'python-gpu/test_gpu_demos.py::test_dask_training',
'python-gpu/test_gpu_prediction.py::TestGPUPredict::test_shap',
'python-gpu/test_gpu_linear.py::TestGPULinear'
]
skip_mark = pytest.mark.skip(reason='This test is not run when --use-rmm-pool flag is active')
for item in items:
if any(item.nodeid.startswith(x) for x in blocklist):
item.add_marker(skip_mark)