Use single precision in gain calculation, use pointers instead of span. (#8051)

This commit is contained in:
Rory Mitchell
2022-07-12 21:56:27 +02:00
committed by GitHub
parent a5bc8e2c6a
commit 0bdaca25ca
3 changed files with 19 additions and 21 deletions

View File

@@ -89,7 +89,7 @@ common::Span<bst_feature_t const> GPUHistEvaluator<GradientSumT>::SortHistogram(
input.gradient_histogram[j]);
return thrust::make_tuple(i, lw);
}
return thrust::make_tuple(i, 0.0);
return thrust::make_tuple(i, 0.0f);
});
// Sort an array segmented according to
// - nodes