From 79ed0ff6e21b1722111ffeb54bd3d13bd5199c86 Mon Sep 17 00:00:00 2001
From: Abhinav Bhatele <bhatele@llnl.gov>
Date: Fri, 21 Jul 2017 16:03:02 -0700
Subject: [PATCH] bug fix for express mesh and dragonfly custom

---
 tracer/tracer-driver.C | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/tracer/tracer-driver.C b/tracer/tracer-driver.C
index 2e676f5..7b460e5 100644
--- a/tracer/tracer-driver.C
+++ b/tracer/tracer-driver.C
@@ -208,6 +208,17 @@ int main(int argc, char **argv)
                 "modelnet_dragonfly_router", NULL, 1);
     }
 
+    if(net_id == DRAGONFLY_CUSTOM) {
+        num_nics = codes_mapping_get_lp_count("MODELNET_GRP", 0,
+                "modelnet_dragonfly_custom", NULL, 1);
+        num_routers = codes_mapping_get_lp_count("MODELNET_GRP", 0,
+                "modelnet_dragonfly_custom_router", NULL, 1);
+        num_nics_per_rep = codes_mapping_get_lp_count("MODELNET_GRP", 1,
+                "modelnet_dragonfly_custom", NULL, 1);
+        num_routers_per_rep = codes_mapping_get_lp_count("MODELNET_GRP", 1,
+                "modelnet_dragonfly_custom_router", NULL, 1);
+    }
+
     if(net_id == FATTREE) {
         num_nics = codes_mapping_get_lp_count("MODELNET_GRP", 0,
                 "modelnet_fattree", NULL, 1);
@@ -230,6 +241,17 @@ int main(int argc, char **argv)
                 "slimfly_router", NULL, 1);
     }
 
+    if(net_id == EXPRESS_MESH) {
+        num_nics = codes_mapping_get_lp_count("MODELNET_GRP", 0,
+                "modelnet_express_mesh", NULL, 1);
+        num_routers = codes_mapping_get_lp_count("MODELNET_GRP", 0,
+                "modelnet_express_mesh_router", NULL, 1);
+        num_nics_per_rep = codes_mapping_get_lp_count("MODELNET_GRP", 1,
+                "modelnet_express_mesh", NULL, 1);
+        num_routers_per_rep = codes_mapping_get_lp_count("MODELNET_GRP", 1,
+                "modelnet_express_mesh_router", NULL, 1);
+    }
+
     num_servers_per_rep = codes_mapping_get_lp_count("MODELNET_GRP", 1,
         "server", NULL, 1);