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

fix typo in arguments of PartitionBuilder::Init (#7113)

Browse Source 
Co-authored-by: Yuntian Zhang <zhangyt@lamda.nju.edu.cn>
This commit is contained in:
farfarawayzyt 2021-07-16 15:46:22 +08:00 committed by GitHub
parent bd1f3a38f0
commit d7c14496d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/partition_builder.h
View File

@ -28,13 +28,13 @@ template<size_t BlockSize>
class PartitionBuilder { class PartitionBuilder {
public: public:
template<typename Func> template<typename Func>
void Init(const size_t n_tasks, size_t n_nodes, Func funcNTaks) { void Init(const size_t n_tasks, size_t n_nodes, Func funcNTask) {
left_right_nodes_sizes_.resize(n_nodes); left_right_nodes_sizes_.resize(n_nodes);
blocks_offsets_.resize(n_nodes+1); blocks_offsets_.resize(n_nodes+1);
blocks_offsets_[0] = 0; blocks_offsets_[0] = 0;
for (size_t i = 1; i < n_nodes+1; ++i) { for (size_t i = 1; i < n_nodes+1; ++i) {
blocks_offsets_[i] = blocks_offsets_[i-1] + funcNTaks(i-1); blocks_offsets_[i] = blocks_offsets_[i-1] + funcNTask(i-1);
} }
if (n_tasks > max_n_tasks_) { if (n_tasks > max_n_tasks_) {