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

Expose DMatrix API for CUDA columnar and array. (#7217)

Browse Source 
* Use JSON encoded configurations.
* Expose them into header file.
This commit is contained in:
Jiaming Yuan
2021-09-09 17:55:25 +08:00
committed by GitHub
parent 68a2c7b8d6
commit 804b2ac60f
4 changed files with 57 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
python-package/xgboost/data.py
View File

@@ -421,17 +421,17 @@ def _transform_cudf_df(data, feature_names, feature_types, enable_categorical):
def _from_cudf_df(
data, missing, nthread, feature_names, feature_types, enable_categorical
):
) -> Tuple[ctypes.c_void_p, Any, Any]:
data, feature_names, feature_types = _transform_cudf_df(
data, feature_names, feature_types, enable_categorical
)
_, interfaces_str = _cudf_array_interfaces(data)
handle = ctypes.c_void_p()
config = bytes(json.dumps({"missing": missing, "nthread": nthread}), "utf-8")
_check_call(
_LIB.XGDMatrixCreateFromArrayInterfaceColumns(
_LIB.XGDMatrixCreateFromCudaColumnar(
interfaces_str,
ctypes.c_float(missing),
ctypes.c_int(nthread),
config,
ctypes.byref(handle),
)
)
@@ -469,11 +469,11 @@ def _from_cupy_array(data, missing, nthread, feature_names, feature_types):
data = _transform_cupy_array(data)
interface_str = _cuda_array_interface(data)
handle = ctypes.c_void_p()
config = bytes(json.dumps({"missing": missing, "nthread": nthread}), "utf-8")
_check_call(
_LIB.XGDMatrixCreateFromArrayInterface(
_LIB.XGDMatrixCreateFromCudaArrayInterface(
interface_str,
ctypes.c_float(missing),
ctypes.c_int(nthread),
config,
ctypes.byref(handle)))
return handle, feature_names, feature_types