support HIP for half in coll

2023-11-02 10:53:12 -07:00 · 2023-11-02 10:53:12 -07:00 · 51efb7442e
commit 51efb7442e
parent 3af5dfd546
1 changed files with 2 additions and 0 deletions
--- a/src/collective/coll.cc
+++ b/src/collective/coll.cc
@ -25,6 +25,8 @@ template <typename T>
 bool constexpr IsFloatingPointV() {
 #if defined(XGBOOST_USE_CUDA)
  return std::is_floating_point_v<T> || std::is_same_v<T, __half>;
+#elif defined(XGBOOST_USE_HIP) /* hack for HIP/Clang */
+  return std::is_floating_point_v<T> || (sizeof(T) == sizeof(unsigned short));
 #else
  return std::is_floating_point_v<T>;
 #endif  // defined(XGBOOST_USE_CUDA)