Updates to GPUTreeShap (#6087)

* Extract paths on device * Update GPUTreeShap
2020-09-06 13:39:08 +12:00
parent 0e2d5669f6
commit 2e907abdb8
4 changed files with 110 additions and 52 deletions
--- a/src/common/host_device_vector.cu
+++ b/src/common/host_device_vector.cu
@@ -404,6 +404,7 @@ template class HostDeviceVector<Entry>;
 template class HostDeviceVector<uint64_t>;  // bst_row_t
 template class HostDeviceVector<uint32_t>;  // bst_feature_t
 template class HostDeviceVector<RegTree::Node>;
+template class HostDeviceVector<RTreeNodeStat>;

 #if defined(__APPLE__)
 /*