cf6137b530
Bring Slurm PMI-1 component online Bring the s2 component online Little cleanup - let the various PMIx modules set the process name during init, and then just raise it up to the ORTE level. Required as the different PMI environments all pass the jobid in different ways. Bring the OMPI pubsub/pmi component online Get comm_spawn working again Ensure we always provide a cpuset, even if it is NULL pmix/cray: adjust cray pmix component for pmix Make changes so cray pmix can work within the integrated ompi/pmix framework. Bring singletons back online. Implement the comm_spawn operation using pmix - not tested yet Cleanup comm_spawn - procs now starting, error in connect_accept Complete integration
146 строки
5.1 KiB
C
146 строки
5.1 KiB
C
/*
|
|
* Copyright (c) 2012-2013 Los Alamos National Security, LLC.
|
|
* All rights reserved.
|
|
* Copyright (c) 2013-2015 Intel, Inc. All rights reserved
|
|
* Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
|
|
* Copyright (c) 2014 Research Organization for Information Science
|
|
* and Technology (RIST). All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*
|
|
* When this component is used, this file is included in the rest of
|
|
* the OPAL/ORTE/OMPI code base via ompi/mca/rte/rte.h. As such,
|
|
* this header represents the public interface to this static component.
|
|
*/
|
|
|
|
#ifndef MCA_OMPI_RTE_ORTE_H
|
|
#define MCA_OMPI_RTE_ORTE_H
|
|
|
|
#include "ompi_config.h"
|
|
#include "ompi/constants.h"
|
|
|
|
struct opal_proc_t;
|
|
|
|
#include "opal/threads/threads.h"
|
|
|
|
#include "orte/types.h"
|
|
#include "orte/mca/errmgr/errmgr.h"
|
|
#include "orte/mca/rml/base/rml_contact.h"
|
|
#include "orte/mca/rml/rml.h"
|
|
#include "orte/mca/routed/routed.h"
|
|
#include "orte/runtime/orte_data_server.h"
|
|
#include "orte/runtime/runtime.h"
|
|
#include "orte/util/name_fns.h"
|
|
#include "orte/util/proc_info.h"
|
|
|
|
#include "ompi/info/info.h"
|
|
struct ompi_proc_t;
|
|
struct ompi_communicator_t;
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
/* Process name objects and operations */
|
|
typedef orte_process_name_t ompi_process_name_t;
|
|
typedef orte_jobid_t ompi_jobid_t;
|
|
typedef orte_vpid_t ompi_vpid_t;
|
|
typedef orte_ns_cmp_bitmask_t ompi_rte_cmp_bitmask_t;
|
|
#define OMPI_PROC_MY_NAME ORTE_PROC_MY_NAME
|
|
#define OMPI_NAME_PRINT(a) ORTE_NAME_PRINT((const orte_process_name_t*)a)
|
|
#define ompi_rte_compare_name_fields(a, b, c) orte_util_compare_name_fields(a, (const orte_process_name_t*)(b), (const orte_process_name_t*)(c))
|
|
#define ompi_rte_convert_string_to_process_name(a,b) orte_util_convert_string_to_process_name(a,b)
|
|
#define ompi_rte_convert_process_name_to_string(a,b) orte_util_convert_process_name_to_string(a,b)
|
|
#define OMPI_NAME_WILDCARD ORTE_NAME_WILDCARD
|
|
#define OMPI_NODE_RANK_INVALID ORTE_NODE_RANK_INVALID
|
|
#define OMPI_LOCAL_RANK_INVALID ORTE_LOCAL_RANK_INVALID
|
|
#define OMPI_RTE_CMP_JOBID ORTE_NS_CMP_JOBID
|
|
#define OMPI_RTE_CMP_VPID ORTE_NS_CMP_VPID
|
|
#define OMPI_RTE_CMP_ALL ORTE_NS_CMP_ALL
|
|
/* This is the DSS tag to serialize a proc name */
|
|
#define OMPI_NAME ORTE_NAME
|
|
#define OMPI_PROCESS_NAME_HTON ORTE_PROCESS_NAME_HTON
|
|
#define OMPI_PROCESS_NAME_NTOH ORTE_PROCESS_NAME_NTOH
|
|
#define OMPI_RTE_MY_NODEID ORTE_PROC_MY_DAEMON->vpid
|
|
|
|
/* database keys */
|
|
#define OMPI_RTE_NODE_ID ORTE_DB_DAEMON_VPID
|
|
#define OMPI_RTE_HOST_ID ORTE_DB_HOSTID
|
|
#if OPAL_ENABLE_DEBUG
|
|
static inline orte_process_name_t * OMPI_CAST_RTE_NAME(opal_process_name_t * name);
|
|
#else
|
|
#define OMPI_CAST_RTE_NAME(a) ((orte_process_name_t*)(a))
|
|
#endif
|
|
|
|
/* Process info struct and values */
|
|
typedef orte_node_rank_t ompi_node_rank_t;
|
|
typedef orte_local_rank_t ompi_local_rank_t;
|
|
#define ompi_process_info orte_process_info
|
|
#define ompi_rte_proc_is_bound orte_proc_is_bound
|
|
|
|
/* Error handling objects and operations */
|
|
OMPI_DECLSPEC void __opal_attribute_noreturn__
|
|
ompi_rte_abort(int error_code, char *fmt, ...);
|
|
#define ompi_rte_abort_peers(a, b, c) orte_errmgr.abort_peers(a, b, c)
|
|
#define OMPI_RTE_ERRHANDLER_FIRST ORTE_ERRMGR_CALLBACK_FIRST
|
|
#define OMPI_RTE_ERRHANDLER_LAST ORTE_ERRMGR_CALLBACK_LAST
|
|
#define OMPI_RTE_ERRHANDLER_PREPEND ORTE_ERRMGR_CALLBACK_PREPEND
|
|
#define OMPI_RTE_ERRHANDLER_APPEND ORTE_ERRMGR_CALLBACK_APPEND
|
|
typedef orte_error_t ompi_rte_error_report_t;
|
|
#define ompi_rte_register_errhandler(a, b) orte_errmgr.register_error_callback(a, b)
|
|
#define OMPI_ERROR_LOG ORTE_ERROR_LOG
|
|
|
|
/* Init and finalize objects and operations */
|
|
#define ompi_rte_init(a, b) orte_init(a, b, ORTE_PROC_MPI)
|
|
#define ompi_rte_finalize() orte_finalize()
|
|
OMPI_DECLSPEC void ompi_rte_wait_for_debugger(void);
|
|
|
|
#define OMPI_DB_HOSTNAME ORTE_DB_HOSTNAME
|
|
#define OMPI_DB_LOCALITY ORTE_DB_LOCALITY
|
|
#define OMPI_DB_GLOBAL_RANK ORTE_DB_GLOBAL_RANK
|
|
|
|
/* Communications */
|
|
typedef orte_rml_tag_t ompi_rml_tag_t;
|
|
#define ompi_rte_send_buffer_nb(a, b, c, d, e) orte_rml.send_buffer_nb(a, b, c, d, e)
|
|
#define ompi_rte_recv_buffer_nb(a, b, c, d, e) orte_rml.recv_buffer_nb(a, b, c, d, e)
|
|
#define ompi_rte_recv_cancel(a, b) orte_rml.recv_cancel(a, b)
|
|
#define ompi_rte_parse_uris(a, b, c) orte_rml_base_parse_uris(a, b, c)
|
|
#define ompi_rte_send_cbfunc orte_rml_send_callback
|
|
|
|
/* Communication tags */
|
|
/* carry over the INVALID def */
|
|
#define OMPI_RML_TAG_INVALID ORTE_RML_TAG_INVALID
|
|
/* define a starting point to avoid conflicts */
|
|
#define OMPI_RML_TAG_BASE ORTE_RML_TAG_MAX
|
|
|
|
#define OMPI_RML_PERSISTENT ORTE_RML_PERSISTENT
|
|
#define OMPI_RML_NON_PERSISTENT ORTE_RML_NON_PERSISTENT
|
|
|
|
typedef struct {
|
|
ompi_rte_component_t super;
|
|
opal_mutex_t lock;
|
|
opal_list_t modx_reqs;
|
|
} ompi_rte_orte_component_t;
|
|
|
|
typedef struct {
|
|
opal_list_item_t super;
|
|
opal_mutex_t lock;
|
|
opal_condition_t cond;
|
|
bool active;
|
|
orte_process_name_t peer;
|
|
} ompi_orte_tracker_t;
|
|
OBJ_CLASS_DECLARATION(ompi_orte_tracker_t);
|
|
|
|
#if OPAL_ENABLE_DEBUG
|
|
static inline orte_process_name_t * OMPI_CAST_RTE_NAME(opal_process_name_t * name) {
|
|
return (orte_process_name_t *)name;
|
|
}
|
|
#endif
|
|
|
|
#define ompi_direct_modex_cutoff orte_direct_modex_cutoff
|
|
|
|
END_C_DECLS
|
|
|
|
#endif /* MCA_OMPI_RTE_ORTE_H */
|