MCPcopy Create free account
hub / github.com/dmlc/xgboost / _transform_cupy_array

Function _transform_cupy_array

python-package/xgboost/data.py:1174–1181  ·  view source on GitHub ↗
(data: DataType)

Source from the content-addressed store, hash-verified

1172
1173
1174def _transform_cupy_array(data: DataType) -> CupyT:
1175 import cupy
1176
1177 if not hasattr(data, "__cuda_array_interface__") and hasattr(data, "__array__"):
1178 data = cupy.array(data, copy=False)
1179 if array_hasobject(data) or data.dtype in [cupy.bool_]:
1180 data = data.astype(cupy.float32, copy=False)
1181 return data
1182
1183
1184def _from_cupy_array(

Callers 4

inplace_predictMethod · 0.85
_from_cupy_arrayFunction · 0.85
_meta_from_cupy_arrayFunction · 0.85
_proxy_transformFunction · 0.85

Calls 1

array_hasobjectFunction · 0.85

Tested by

no test coverage detected