Implement host span. (#5459)

2020-04-03 10:37:51 +08:00
parent 459b175dc6
commit 86beb68ce8
2 changed files with 12 additions and 0 deletions
--- a/include/xgboost/host_device_vector.h
+++ b/include/xgboost/host_device_vector.h
@@ -105,6 +105,9 @@ class HostDeviceVector {
  const T* DevicePointer() const { return ConstDevicePointer(); }

  T* HostPointer() { return HostVector().data(); }
+  common::Span<T> HostSpan() { return common::Span<T>{HostVector()}; }
+  common::Span<T const> HostSpan() const { return common::Span<T const>{HostVector()}; }
+  common::Span<T const> ConstHostSpan() const { return HostSpan(); }
  const T* ConstHostPointer() const { return ConstHostVector().data(); }
  const T* HostPointer() const { return ConstHostPointer(); }