Skip to content

Commit

Permalink
Fix dask interface with latest cupy. (#8210)
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Sep 2, 2022
1 parent deae99e commit ada4a86
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions python-package/xgboost/compat.py
Expand Up @@ -4,7 +4,7 @@
import logging
import sys
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

Expand Down Expand Up @@ -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

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

# pylint: disable=c-extension-no-member,no-member
Expand Down

0 comments on commit ada4a86

Please sign in to comment.