From 332d30cedc2179b93305a6d2011b83b6e7e0a2c0 Mon Sep 17 00:00:00 2001
From: Christian Heinrich <franz-christian.heinrich@inria.fr>
Date: Tue, 17 Jul 2018 13:38:39 +0200
Subject: [PATCH] [SMPI/LB] Use mapping class within the LB

---
 include/simgrid/smpi/loadbalancer/load_balancer.hpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/include/simgrid/smpi/loadbalancer/load_balancer.hpp b/include/simgrid/smpi/loadbalancer/load_balancer.hpp
index 37606ba54d..6bc4e72939 100644
--- a/include/simgrid/smpi/loadbalancer/load_balancer.hpp
+++ b/include/simgrid/smpi/loadbalancer/load_balancer.hpp
@@ -2,6 +2,8 @@
 #ifndef HAVE_SG_PLUGIN_LB
 #define HAVE_SG_PLUGIN_LB
 
+#include <simgrid/s4u.hpp>
+
 namespace simgrid {
 namespace plugin {
 namespace loadbalancer {
@@ -52,8 +54,8 @@ public:
 
 class LoadBalancer
 {
-  std::map<int, double> actor_computation;
-  std::map<simgrid::s4u::ActorPtr, simgrid::s4u::Host*> new_mapping;
+  Mapping new_mapping;
+  std::map</*proc id*/int, double> actor_computation;
 
 public:
   LoadBalancer();
-- 
2.20.1