From abdf894fcf371a1f1cf00eb8425f890e8aee1825 Mon Sep 17 00:00:00 2001 From: Rory Mitchell Date: Thu, 18 Jun 2020 16:55:09 +1200 Subject: [PATCH] Add cupy to Windows CI (#5797) * Add cupy to Windows CI * Update Jenkinsfile-win64 Co-authored-by: Philip Hyunsu Cho * Update Jenkinsfile-win64 Co-authored-by: Philip Hyunsu Cho * Update tests/python-gpu/test_gpu_prediction.py Co-authored-by: Philip Hyunsu Cho Co-authored-by: Philip Hyunsu Cho --- Jenkinsfile-win64 | 3 ++- tests/python-gpu/test_gpu_prediction.py | 3 +++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/Jenkinsfile-win64 b/Jenkinsfile-win64 index 9233494d3..eba5464ff 100644 --- a/Jenkinsfile-win64 +++ b/Jenkinsfile-win64 @@ -137,8 +137,9 @@ def TestWin64GPU(args) { conda activate && for /R %%i in (python-package\\dist\\*.whl) DO python -m pip install "%%i" """ echo "Installing Python dependencies..." + def cuda_short_ver = args.cuda_target.replaceAll('_', '') bat """ - conda activate && conda install -y hypothesis && conda upgrade scikit-learn pandas numpy hypothesis + conda activate && conda install -y hypothesis && conda upgrade scikit-learn pandas numpy hypothesis && python -m pip install cupy-${cuda_short_ver} """ echo "Running Python tests..." bat """ diff --git a/tests/python-gpu/test_gpu_prediction.py b/tests/python-gpu/test_gpu_prediction.py index 6324e117b..26ae95d1b 100644 --- a/tests/python-gpu/test_gpu_prediction.py +++ b/tests/python-gpu/test_gpu_prediction.py @@ -147,6 +147,9 @@ class TestGPUPredict(unittest.TestCase): copied_predt = cp.array(booster.predict(d)) return cp.all(copied_predt == inplace_predt) + # Don't do this on Windows, see issue #5793 + if sys.platform.startswith("win"): + pytest.skip('Multi-threaded in-place prediction with cuPy is not working on Windows') for i in range(10): run_threaded_predict(X, rows, predict_dense)