Implement devices to devices reshard. (#3721)

* Force clearing device memory before Reshard.
* Remove calculating row_segments for gpu_hist and gpu_sketch.
* Guard against changing device.
This commit is contained in:
trivialfis
2018-09-28 17:40:23 +12:00
committed by Rory Mitchell
parent 0b7fd74138
commit 5a7f7e7d49
11 changed files with 179 additions and 96 deletions

View File

@@ -67,7 +67,7 @@ struct HistCutUnit {
: cut(cut), size(size) {}
};
/*! \brief cut configuration for all the features */
/*! \brief cut configuration for all the features. */
struct HistCutMatrix {
/*! \brief unit pointer to rows by element position */
std::vector<uint32_t> row_ptr;