Allow compilation with nvcc 11.4 (#7131)
* Use type aliases for discard iterators * update to include host_vector as thrust 1.12 doesn't bring it in as a side-effect * cub::DispatchRadixSort requires signed offset types
This commit is contained in:
@@ -20,10 +20,7 @@ namespace xgboost {
|
||||
namespace metric {
|
||||
namespace {
|
||||
template <typename T>
|
||||
class Discard : public thrust::discard_iterator<T> {
|
||||
public:
|
||||
using value_type = T; // NOLINT
|
||||
};
|
||||
using Discard = thrust::discard_iterator<T>;
|
||||
|
||||
struct GetWeightOp {
|
||||
common::Span<float const> weights;
|
||||
|
||||
Reference in New Issue
Block a user