Refactor split valuation kernel (#8073)

This commit is contained in:
Rory Mitchell
2022-07-21 15:41:50 +02:00
committed by GitHub
parent cb40bbdadd
commit 1be09848a7
9 changed files with 236 additions and 233 deletions

View File

@@ -1949,7 +1949,7 @@ class LDGIterator {
const T *ptr_;
public:
explicit LDGIterator(const T *ptr) : ptr_(ptr) {}
XGBOOST_DEVICE explicit LDGIterator(const T *ptr) : ptr_(ptr) {}
__device__ T operator[](std::size_t idx) const {
DeviceWordT tmp[kNumWords];
static_assert(sizeof(tmp) == sizeof(T), "Expect sizes to be equal.");