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

Add DaskDeviceQuantileDMatrix demo. (#6156)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-09-24 14:08:28 +08:00
committed by GitHub
parent 678ea40b24
commit 78d72ef936
3 changed files with 63 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/python-gpu/test_gpu_demos.py
View File

@@ -6,8 +6,22 @@ sys.path.append("tests/python")
import testing as tm
import test_demos as td # noqa
@pytest.mark.skipif(**tm.no_cupy())
def test_data_iterator():
script = os.path.join(td.PYTHON_DEMO_DIR, 'data_iterator.py')
cmd = ['python', script]
subprocess.check_call(cmd)
@pytest.mark.skipif(**tm.no_dask())
@pytest.mark.skipif(**tm.no_dask_cuda())
@pytest.mark.skipif(**tm.no_cupy())
@pytest.mark.mgpu
def test_dask_training():
script = os.path.join(tm.PROJECT_ROOT, 'demo', 'dask', 'gpu_training.py')
cmd = ['python', script, '--ddqdm=1']
subprocess.check_call(cmd)
cmd = ['python', script, '--ddqdm=0']
subprocess.check_call(cmd)