Avoid thrust vector initialization. (#10544)
* Avoid thrust vector initialization. - Add a wrapper for rmm device uvector. - Split up the `Resize` method for HDV.
This commit is contained in:
@@ -135,7 +135,9 @@ class HostDeviceVector {
|
||||
|
||||
void SetDevice(DeviceOrd device) const;
|
||||
|
||||
void Resize(size_t new_size, T v = T());
|
||||
void Resize(std::size_t new_size);
|
||||
/** @brief Resize and initialize the data if the new size is larger than the old size. */
|
||||
void Resize(std::size_t new_size, T v);
|
||||
|
||||
using value_type = T; // NOLINT
|
||||
|
||||
|
||||
Reference in New Issue
Block a user