#1395: lb: add hash for new enum

DARMA-tasking · Aug 23, 2021 · 79efa85 · 79efa85
1 parent c1ad7c7
commit 79efa85
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/greedylb/greedylb.h b/src/vt/vrt/collection/balance/greedylb/greedylb.h
@@ -124,4 +124,18 @@ struct GreedyLB : BaseLB {
 
 }}}} /* end namespace vt::vrt::collection::lb */
 
+namespace std {
+
+template <>
+struct hash<::vt::vrt::collection::lb::DataDistStrategy> {
+  size_t operator()(::vt::vrt::collection::lb::DataDistStrategy const& in) const {
+    using under = std::underlying_type<
+      ::vt::vrt::collection::lb::DataDistStrategy
+    >::type;
+    return std::hash<under>()(static_cast<under>(in));
+  }
+};
+
+} /* end namespace std */
+
 #endif /*INCLUDED_VT_VRT_COLLECTION_BALANCE_GREEDYLB_GREEDYLB_H*/