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
src/c_api/c_api.cc
View File

@@ -169,19 +169,17 @@ XGB_DLL int XGDMatrixCreateFromDataIter(
}
#ifndef XGBOOST_USE_CUDA
XGB_DLL int XGDMatrixCreateFromArrayInterfaceColumns(char const* c_json_strs,
bst_float missing,
int nthread,
DMatrixHandle* out) {
XGB_DLL int XGDMatrixCreateFromCudaColumnar(char const *data,
char const* c_json_config,
DMatrixHandle *out) {
API_BEGIN();
common::AssertGPUSupport();
API_END();
}
XGB_DLL int XGDMatrixCreateFromArrayInterface(char const* c_json_strs,
bst_float missing,
int nthread,
DMatrixHandle* out) {
XGB_DLL int XGDMatrixCreateFromCudaArrayInterface(char const *data,
char const* c_json_config,
DMatrixHandle *out) {
API_BEGIN();
common::AssertGPUSupport();
API_END();