diff --git a/src/tree/common_row_partitioner.h b/src/tree/common_row_partitioner.h index 159be768c..1bf185341 100644 --- a/src/tree/common_row_partitioner.h +++ b/src/tree/common_row_partitioner.h @@ -145,9 +145,10 @@ class CommonRowPartitioner { } /* Making GHistIndexMatrix_t a templete parameter allows reuse this function for sycl-plugin */ - template + template static void FindSplitConditions(const std::vector& nodes, const RegTree& tree, - const GHistIndexMatrix_t& gmat, std::vector* split_conditions) { + GHistIndexMatrixT const& gmat, + std::vector* split_conditions) { auto const& ptrs = gmat.cut.Ptrs(); auto const& vals = gmat.cut.Values();