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 15:25:49 +08:00
committed by GitHub
co-authored by Dmitry Razdoburdin
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
src/tree/common_row_partitioner.h
+4 -3
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();