Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[coll] allgatherv. (#9688)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-10-19 03:13:50 +08:00
committed by GitHub
parent ea9f09716b
commit 5d1bcde719
8 changed files with 157 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/common/host_device_vector.cc
View File

@@ -175,6 +175,7 @@ template class HostDeviceVector<GradientPair>;
template class HostDeviceVector<GradientPairPrecise>;
template class HostDeviceVector<int32_t>; // bst_node_t
template class HostDeviceVector<uint8_t>;
template class HostDeviceVector<int8_t>;
template class HostDeviceVector<FeatureType>;
template class HostDeviceVector<Entry>;
template class HostDeviceVector<uint64_t>; // bst_row_t

1
src/common/host_device_vector.cu
View File

@@ -409,6 +409,7 @@ template class HostDeviceVector<GradientPair>;
template class HostDeviceVector<GradientPairPrecise>;
template class HostDeviceVector<int32_t>; // bst_node_t
template class HostDeviceVector<uint8_t>;
template class HostDeviceVector<int8_t>;
template class HostDeviceVector<FeatureType>;
template class HostDeviceVector<Entry>;
template class HostDeviceVector<uint64_t>; // bst_row_t