d9acc93efa
To be precise, given this hypothetical launching pattern: host1: vpids 0, 2, 4, 6 host2: vpids 1, 3, 5, 7 The local_rank for these procs would be: host1: vpids 0->local_rank 0, v2->lr1, v4->lr2, v6->lr3 host2: vpids 1->local_rank 0, v3->lr1, v5->lr2, v7->lr3 and the number of local procs on each node would be four. If vpid=0 then does a comm_spawn of one process on host1, the values of the parent job would remain unchanged. The local_rank of the child process would be 0 and its num_local_procs would be 1 since it is in a separate jobid. I have verified this functionality for the rsh case - need to verify that slurm and other cases also get the right values. Some consolidation of common code is probably going to occur in the SDS components to make this simpler and more maintainable in the future. This commit was SVN r14706.
121 строка
4.5 KiB
C
121 строка
4.5 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2005 The University of Tennessee and The University
|
|
* of Tennessee Research Foundation. All rights
|
|
* reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* Copyright (c) 2004-2005 The Regents of the University of California.
|
|
* All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
/** @file:
|
|
*
|
|
* Populates global structure with process-specific information.
|
|
*
|
|
*
|
|
*/
|
|
|
|
#ifndef _ORTE_PROC_INFO_H_
|
|
#define _ORTE_PROC_INFO_H_
|
|
|
|
#include "orte_config.h"
|
|
|
|
#ifdef HAVE_SYS_TYPES_H
|
|
#include <sys/types.h>
|
|
#endif
|
|
#include "orte/mca/ns/ns_types.h"
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
/**
|
|
* Process information structure
|
|
*
|
|
* The orte_proc_info() function fills the pid field and obtains the
|
|
* process name, storing that information in the global structure. The
|
|
* structure also holds path names to the universe, job, and process
|
|
* session directories, and to the stdin, stdout, and stderr temp
|
|
* files - however, these are all initialized elsewhere.
|
|
*/
|
|
struct orte_proc_info_t {
|
|
orte_process_name_t *my_name; /**< My official process name */
|
|
orte_std_cntr_t app_num; /**< our index into the app_context array */
|
|
bool singleton; /**< I am a singleton */
|
|
orte_vpid_t vpid_start; /**< starting vpid for this job */
|
|
orte_std_cntr_t num_procs; /**< number of processes in this job */
|
|
orte_vpid_t local_rank; /**< local rank on this node */
|
|
orte_std_cntr_t num_local_procs; /**< total number of procs on this node */
|
|
pid_t pid; /**< Local process ID for this process */
|
|
bool seed; /**< Indicate whether or not this is seed daemon */
|
|
bool daemon; /**< Indicate whether or not I am a daemon */
|
|
char *ns_replica_uri; /**< contact info for name services replica */
|
|
char *gpr_replica_uri; /**< contact info for registry replica */
|
|
orte_process_name_t *ns_replica; /**< Name of my name server replica (NULL=>me) */
|
|
orte_process_name_t *gpr_replica; /**< Name of my registry replica (NULL=>me) */
|
|
char *tmpdir_base; /**< Base directory of the session dir tree */
|
|
char *top_session_dir; /**< Top-most directory of the session tree */
|
|
char *universe_session_dir; /**< Location of universe temp dir.
|
|
* The session directory has the form
|
|
* <prefix><openmpi-sessions-user><universe>, where the prefix
|
|
* can either be provided by the user via the
|
|
* --tmpdir command-line flag, the use of one of several
|
|
* environmental variables, or else a default location.
|
|
*/
|
|
|
|
char *job_session_dir; /**< Session directory for job */
|
|
|
|
char *proc_session_dir; /**< Session directory for the process */
|
|
|
|
char *sock_stdin; /**< Path name to temp file for stdin. */
|
|
char *sock_stdout; /**< Path name to temp file for stdout. */
|
|
char *sock_stderr; /**< Path name to temp file for stderr. */
|
|
};
|
|
typedef struct orte_proc_info_t orte_proc_info_t;
|
|
|
|
|
|
/**
|
|
*
|
|
* Global process info descriptor. Initialized to almost no
|
|
* meaningful information - data is provided by calling \c
|
|
* orte_rte_init() (which calls \c orte_proc_info() to fill in the
|
|
* structure).
|
|
*
|
|
* The exception to this rule is the \c orte_process_info.seed field,
|
|
* which will be initialized to \c false, but should be set to \c true
|
|
* before calling \c orte_rte_info() if the caller is a seed daemon.
|
|
*/
|
|
ORTE_DECLSPEC extern orte_proc_info_t orte_process_info;
|
|
|
|
|
|
/**
|
|
* \internal
|
|
*
|
|
* Global structure to store a wide range of information about the
|
|
* process. orte_proc_info populates a global variable with
|
|
* information about the process being executing. This function should
|
|
* be called only once, from orte_rte_init().
|
|
*
|
|
* @param None.
|
|
*
|
|
* @retval ORTE_SUCCESS Successfully initialized the various fields.
|
|
* @retval OMPI_ERROR Failed to initialize one or more fields.
|
|
*/
|
|
|
|
ORTE_DECLSPEC int orte_proc_info(void);
|
|
|
|
ORTE_DECLSPEC int orte_proc_info_finalize(void);
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif
|