Support dmatrix construction from cupy array (#5206)

2020-01-22 13:15:27 +13:00
parent 2a071cebc5
commit 9c56480c61
19 changed files with 522 additions and 158 deletions
--- a/tests/python/testing.py
+++ b/tests/python/testing.py
@@ -48,6 +48,15 @@ def no_cudf():
            'reason': 'CUDF is not installed'}


+def no_cupy():
+    reason = 'cupy is not installed.'
+    try:
+        import cupy as _   # noqa
+        return {'condition': False, 'reason': reason}
+    except ImportError:
+        return {'condition': True, 'reason': reason}
+
+
 def no_dask_cudf():
    reason = 'dask_cudf is not installed.'
    try: