bug fix for express mesh and dragonfly custom

hpcgroup · Jul 22, 2017 · 79ed0ff · 79ed0ff
1 parent 5d24f01
commit 79ed0ff
Showing 1 changed file with 22 additions and 0 deletions.
diff --git 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);