6d6cebb4a7
Accordingly, there are new APIs to the name service to support the ability to get a job's parent, root, immediate children, and all its descendants. In addition, the terminate_job, terminate_orted, and signal_job APIs for the PLS have been modified to accept attributes that define the extent of their actions. For example, doing a "terminate_job" with an attribute of ORTE_NS_INCLUDE_DESCENDANTS will terminate the given jobid AND all jobs that descended from it. I have tested this capability on a MacBook under rsh, Odin under SLURM, and LANL's Flash (bproc). It worked successfully on non-MPI jobs (both simple and including a spawn), and MPI jobs (again, both simple and with a spawn). This commit was SVN r12597.
151 строка
4.0 KiB
C
151 строка
4.0 KiB
C
/* -*- C -*-
|
|
*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
|
|
* University Research and Technology
|
|
* Corporation. All rights reserved.
|
|
* Copyright (c) 2004-2006 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$
|
|
*
|
|
*/
|
|
#ifndef NS_PROXY_H
|
|
#define NS_PROXY_H
|
|
|
|
|
|
#include "orte_config.h"
|
|
#include "orte/orte_constants.h"
|
|
|
|
#include "opal/types.h"
|
|
#include "opal/class/opal_list.h"
|
|
|
|
#include "orte/dss/dss.h"
|
|
|
|
#include "orte/mca/ns/ns.h"
|
|
#include "orte/mca/ns/base/ns_private.h"
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
struct orte_ns_proxy_tagitem_t {
|
|
opal_object_t super;
|
|
orte_rml_tag_t tag; /**< OOB tag */
|
|
char *name; /**< Name associated with tag */
|
|
};
|
|
typedef struct orte_ns_proxy_tagitem_t orte_ns_proxy_tagitem_t;
|
|
|
|
OBJ_CLASS_DECLARATION(orte_ns_proxy_tagitem_t);
|
|
|
|
struct orte_ns_proxy_dti_t {
|
|
opal_object_t super;
|
|
orte_data_type_t id; /**< data type id */
|
|
char *name; /**< Name associated with data type */
|
|
};
|
|
typedef struct orte_ns_proxy_dti_t orte_ns_proxy_dti_t;
|
|
|
|
OBJ_CLASS_DECLARATION(orte_ns_proxy_dti_t);
|
|
|
|
|
|
/*
|
|
* Module open / close
|
|
*/
|
|
int orte_ns_proxy_open(void);
|
|
int orte_ns_proxy_close(void);
|
|
|
|
|
|
/*
|
|
* Startup / Shutdown
|
|
*/
|
|
mca_ns_base_module_t* orte_ns_proxy_init(int *priority);
|
|
int orte_ns_proxy_module_init(void);
|
|
int orte_ns_proxy_finalize(void);
|
|
|
|
/*
|
|
* globals used within proxy component
|
|
*/
|
|
typedef struct {
|
|
size_t max_size, block_size;
|
|
int debug;
|
|
orte_cellid_t num_cells;
|
|
orte_pointer_array_t *cells;
|
|
orte_pointer_array_t *tags;
|
|
orte_rml_tag_t num_tags;
|
|
orte_pointer_array_t *dts;
|
|
orte_data_type_t num_dts;
|
|
opal_mutex_t mutex;
|
|
} orte_ns_proxy_globals_t;
|
|
|
|
extern orte_ns_proxy_globals_t orte_ns_proxy;
|
|
|
|
/*
|
|
* simplifying define
|
|
*/
|
|
#define ORTE_NS_MY_REPLICA orte_process_info.ns_replica
|
|
|
|
|
|
/*
|
|
* proxy function prototypes
|
|
*/
|
|
int orte_ns_proxy_create_cellid(orte_cellid_t *cellid, char *site, char *resource);
|
|
|
|
int orte_ns_proxy_get_cell_info(orte_cellid_t cellid, char **site, char **resource);
|
|
|
|
int orte_ns_proxy_create_nodeids(orte_nodeid_t **nodeids, orte_std_cntr_t *nnodes,
|
|
orte_cellid_t cellid, char **nodenames);
|
|
|
|
int orte_ns_proxy_get_node_info(char ***nodename, orte_cellid_t cellid, orte_std_cntr_t num_nodes, orte_nodeid_t *nodeids);
|
|
|
|
int orte_ns_proxy_create_jobid(orte_jobid_t *jobid, opal_list_t *attrs);
|
|
|
|
int orte_ns_proxy_get_job_descendants(orte_jobid_t** descendants, orte_std_cntr_t *ndesc, orte_jobid_t job);
|
|
|
|
int orte_ns_proxy_get_job_children(orte_jobid_t** descendants, orte_std_cntr_t *ndesc, orte_jobid_t job);
|
|
|
|
int orte_ns_proxy_get_root_job(orte_jobid_t *root_job, orte_jobid_t job);
|
|
|
|
int orte_ns_proxy_get_parent_job(orte_jobid_t *parent, orte_jobid_t job);
|
|
|
|
int orte_ns_proxy_reserve_range(orte_jobid_t job, orte_vpid_t range,
|
|
orte_vpid_t *startvpid);
|
|
|
|
int orte_ns_proxy_get_peers(orte_process_name_t **procs,
|
|
orte_std_cntr_t *num_procs, opal_list_t *attrs);
|
|
|
|
int orte_ns_proxy_assign_rml_tag(orte_rml_tag_t *tag, char *name);
|
|
|
|
int orte_ns_proxy_define_data_type(const char *name,
|
|
orte_data_type_t *type);
|
|
|
|
int orte_ns_proxy_create_my_name(void);
|
|
|
|
/*
|
|
* Diagnostic functions
|
|
*/
|
|
int orte_ns_proxy_dump_cells(void);
|
|
|
|
int orte_ns_proxy_dump_jobs(void);
|
|
|
|
int orte_ns_proxy_dump_tags(void);
|
|
|
|
int orte_ns_proxy_dump_datatypes(void);
|
|
|
|
/*
|
|
*
|
|
*/
|
|
ORTE_MODULE_DECLSPEC extern mca_ns_base_component_t mca_ns_proxy_component;
|
|
|
|
#if defined(c_plusplus) || defined(__cplusplus)
|
|
}
|
|
#endif
|
|
|
|
#endif
|