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

[sycl] add split applications and tests (#10636)

Browse Source 
Co-authored-by: Dmitry Razdoburdin <>
This commit is contained in:
Dmitry Razdoburdin
2024-07-26 09:25:49 +02:00
committed by GitHub
parent 384983ed27
commit 7720272870
4 changed files with 180 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/tree/common_row_partitioner.h
View File

@@ -144,9 +144,10 @@ class CommonRowPartitioner {
}
}
template <typename ExpandEntry>
void FindSplitConditions(const std::vector<ExpandEntry>& nodes, const RegTree& tree,
const GHistIndexMatrix& gmat, std::vector<bst_bin_t>* split_conditions) {
/* Making GHistIndexMatrix_t a templete parameter allows reuse this function for sycl-plugin */
template <typename ExpandEntry, typename GHistIndexMatrix_t>
static void FindSplitConditions(const std::vector<ExpandEntry>& nodes, const RegTree& tree,
const GHistIndexMatrix_t& gmat, std::vector<int32_t>* split_conditions) {
auto const& ptrs = gmat.cut.Ptrs();
auto const& vals = gmat.cut.Values();