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
1 changed files with 4 additions and 2 deletions
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