Use new matrix table get interface in c_api.

This commit is contained in:
huxuan 2016-05-24 14:00:21 +09:00
Родитель 849bd59861
Коммит e34d745610
2 изменённых файлов: 6 добавлений и 6 удалений

Просмотреть файл

@ -40,8 +40,9 @@ DllExport void MV_GetMatrixTableAll(TableHandler handler, float* data, int size)
DllExport void MV_AddMatrixTableAll(TableHandler handler, float* data, int size);
DllExport void MV_GetMatrixTableByRows(TableHandler handler, int row_ids[],
int row_ids_n, int num_col, float* data[]);
DllExport void MV_GetMatrixTableByRows(TableHandler handler, float* data,
int num_col, int row_ids[],
int row_ids_n);
DllExport void MV_AddMatrixTableByRows(TableHandler handler, float* data,
int num_col, int row_ids[],

Просмотреть файл

@ -69,11 +69,10 @@ void MV_AddMatrixTableAll(TableHandler handler, float* data, int size) {
worker->Add(data, size);
}
void MV_GetMatrixTableByRows(TableHandler handler, int row_ids[],
int row_ids_n, int num_col, float** data) {
void MV_GetMatrixTableByRows(TableHandler handler, float* data, int num_col,
int row_ids[], int row_ids_n) {
auto worker = reinterpret_cast<multiverso::MatrixWorkerTable<float>*>(handler);
worker->Get(std::vector<multiverso::integer_t>(row_ids, row_ids + row_ids_n),
std::vector<float*>(data, data + row_ids_n), num_col);
worker->Get(data, num_col, row_ids, row_ids_n);
}
void MV_AddMatrixTableByRows(TableHandler handler, float* data, int num_col,