#1382: add union in StatisticQuantityMap

DARMA-tasking · Jul 19, 2022 · e4f1dec · e4f1dec
1 parent 7209641
commit e4f1dec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/vt/vrt/collection/balance/lb_common.h b/src/vt/vrt/collection/balance/lb_common.h
@@ -193,7 +193,12 @@ enum struct Statistic : int8_t {
   // InternalEdgesCardinality
 };
 
-using StatisticQuantityMap = std::map<StatisticQuantity, double>;
+using UnionValueType = vt::adt::SafeUnion<
+    double,
+    adt::HistogramApprox<double, int64_t>
+  >;
+
+using StatisticQuantityMap = std::map<StatisticQuantity, UnionValueType>;
 using StatisticMap = std::unordered_map<Statistic, StatisticQuantityMap>;
 
 nlohmann::json jsonifyPhaseStatistics(const StatisticMap &statistics);