Changeset 5641

Timestamp:

10/30/09 19:01:13 (3 weeks ago)

Author:

balaji

Message:

Fix for the MPICH_INTERFACE_HOSTNAME issue (ticket #903). Now Hydra
will automatically pick the hostname specified by the user in the
hostfile as the preferred communication interface. With this, Hydra
should work more correctly as described in
http://wiki.mcs.anl.gov/mpich2/index.php/Using_the_Hydra_Process_Manager#Hydra_with_Non-Ethernet_Networks

Location:

mpich2/trunk/src/pm/hydra/pm/pmiserv

Files:

• pmi_proxy.h (modified) (1 diff)
• pmi_proxy_utils.c (modified) (4 diffs)
• pmi_serv_launch.c (modified) (1 diff)

mpich2/trunk/src/pm/hydra/pm/pmiserv/pmi_proxy.h

@@ -47 +47 @@
     struct {
         int id;
+        char *hostname;
+
         int proxy_core_count;
         int proxy_process_count;

mpich2/trunk/src/pm/hydra/pm/pmiserv/pmi_proxy_utils.c

@@ -36 +36 @@
 
     HYD_pmcd_pmip.local.id = -1;
+    HYD_pmcd_pmip.local.hostname = NULL;
     HYD_pmcd_pmip.local.proxy_core_count = 0;
     HYD_pmcd_pmip.local.proxy_process_count = 0;
@@ -173 +174 @@
         }
 
+        /* Hostname (as specified by the user) */
+        if (!strcmp(*argv, "--hostname")) {
+            argv++;
+            HYD_pmcd_pmip.local.hostname = HYDU_strdup(*argv);
+            continue;
+        }
+
         /* Process count */
         if (!strcmp(*argv, "--proxy-core-count")) {
@@ -414 +422 @@
         HYDU_FREE(HYD_pmcd_pmip.downstream.exit_status);
 
+    if (HYD_pmcd_pmip.local.hostname)
+        HYDU_FREE(HYD_pmcd_pmip.local.hostname);
+
     HYDT_bind_finalize();
 
@@ -607 +618 @@
         }
 
+        /* Set the MPICH_INTERFACE_HOSTNAME based on what the user provided */
+        if (HYD_pmcd_pmip.local.hostname) {
+            status = HYDU_env_create(&env, "MPICH_INTERFACE_HOSTNAME",
+                                     HYD_pmcd_pmip.local.hostname);
+            HYDU_ERR_POP(status, "unable to create env\n");
+
+            status = HYDU_append_env_to_list(*env, &prop_env);
+            HYDU_ERR_POP(status, "unable to add env to list\n");
+        }
+
         for (i = 0; i < exec->proc_count; i++) {
             pmi_id = HYDU_local_to_global_id(process_id,

mpich2/trunk/src/pm/hydra/pm/pmiserv/pmi_serv_launch.c

@@ -230 +230 @@
 
         arg = 0;
+        proxy->exec_launch_info[arg++] = HYDU_strdup("--hostname");
+        proxy->exec_launch_info[arg++] = HYDU_strdup(proxy->hostname);
+
         proxy->exec_launch_info[arg++] = HYDU_strdup("--global-core-count");
         proxy->exec_launch_info[arg++] = HYDU_int_to_str(HYD_handle.global_core_count);