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

* 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

View File

@@ -210,6 +210,36 @@ XGB_DLL int XGDMatrixCreateFromDT(void** data,
DMatrixHandle* out,
int nthread);
/*!
* \brief Create DMatrix from CUDA columnar format. (cuDF)
* \param data Array of JSON encoded __cuda_array_interface__ for each column.
* \param json_config JSON encoded configuration. Required values are:
*
* - missing
* - nthread
*
* \param out created dmatrix
* \return 0 when success, -1 when failure happens
*/
XGB_DLL int XGDMatrixCreateFromCudaColumnar(char const *data,
char const* json_config,
DMatrixHandle *out);
/*!
* \brief Create DMatrix from CUDA array.
* \param data JSON encoded __cuda_array_interface__ for array data.
* \param json_config JSON encoded configuration. Required values are:
*
* - missing
* - nthread
*
* \param out created dmatrix
* \return 0 when success, -1 when failure happens
*/
XGB_DLL int XGDMatrixCreateFromCudaArrayInterface(char const *data,
char const* json_config,
DMatrixHandle *out);
/*
* ========================== Begin data callback APIs =========================
*