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

Remove omp_get_max_threads in data. (#7588)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-01-24 02:44:07 +08:00
committed by GitHub
parent f84291c1e1
commit 5817840858
18 changed files with 97 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/xgboost/data.h
View File

@@ -300,7 +300,7 @@ class SparsePage {
base_rowid = row_id;
}
SparsePage GetTranspose(int num_columns) const;
SparsePage GetTranspose(int num_columns, int32_t n_threads) const;
void SortRows() {
auto ncol = static_cast<bst_omp_uint>(this->Size());