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

Fix dask interface with latest cupy. (#8210)

Browse Source
This commit is contained in:
Jiaming Yuan 2022-09-03 03:10:43 +08:00 committed by GitHub
parent deae99e662
commit ada4a86d1c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
python-package/xgboost/compat.py
View File

@ -4,7 +4,7 @@ import importlib.util
import logging import logging
import sys import sys
import types import types
from typing import Any, Dict, List, Optional, Sequence, Type, cast from typing import Any, Dict, List, Optional, Sequence, cast
import numpy as np import numpy as np
@ -125,7 +125,9 @@ def concat(value: Sequence[_T]) -> _T: # pylint: disable=too-many-return-statem
from cudf import concat as CUDF_concat # pylint: disable=import-error from cudf import concat as CUDF_concat # pylint: disable=import-error
return CUDF_concat(value, axis=0) return CUDF_concat(value, axis=0)
if lazy_isinstance(value[0], "cupy._core.core", "ndarray"): from .data import _is_cupy_array
if _is_cupy_array(value[0]):
import cupy # pylint: disable=import-error import cupy # pylint: disable=import-error
# pylint: disable=c-extension-no-member,no-member # pylint: disable=c-extension-no-member,no-member