65ba6af44d
Have each mapper flag it did the map so we can see who did it later. Ensure procs are flagged as "ready to launch". This commit was SVN r24406.
109 строки
4.3 KiB
C
109 строки
4.3 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:
|
|
*/
|
|
|
|
#ifndef ORTE_MCA_RMAPS_PRIVATE_H
|
|
#define ORTE_MCA_RMAPS_PRIVATE_H
|
|
|
|
/*
|
|
* includes
|
|
*/
|
|
#include "orte_config.h"
|
|
#include "orte/types.h"
|
|
|
|
#include "orte/runtime/orte_globals.h"
|
|
|
|
#include "orte/mca/rmaps/rmaps.h"
|
|
|
|
BEGIN_C_DECLS
|
|
|
|
/*
|
|
* Base API functions
|
|
*/
|
|
|
|
/*
|
|
* Map a job
|
|
* All calls to rmaps.map_job are routed through this function. This allows callers to
|
|
* the RMAPS framework to specify the particular mapper they wish to use.
|
|
*/
|
|
ORTE_DECLSPEC int orte_rmaps_base_map_job(orte_job_t *jdata);
|
|
ORTE_DECLSPEC orte_job_map_t* orte_rmaps_base_get_job_map(orte_jobid_t job);
|
|
|
|
|
|
/* LOCAL FUNCTIONS for use by RMAPS components */
|
|
|
|
/*
|
|
* Function to add a mapped_proc entry to a map
|
|
* Scans list of nodes on map to see if the specified one already
|
|
* exists - if so, just add this entry to that node's list of
|
|
* procs. If not, then add new node entry and put this proc
|
|
* on its list.
|
|
*/
|
|
int orte_rmaps_base_add_proc_to_map(orte_job_map_t *map, orte_node_t *node,
|
|
bool oversubscribed, orte_proc_t *proc);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_get_target_nodes(opal_list_t* node_list,
|
|
orte_std_cntr_t *total_num_slots,
|
|
orte_app_context_t *app,
|
|
orte_mapping_policy_t policy);
|
|
ORTE_DECLSPEC int orte_rmaps_base_get_target_procs(opal_list_t *procs);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_update_node_usage(opal_list_t *nodes);
|
|
ORTE_DECLSPEC int orte_rmaps_base_get_mapped_targets(opal_list_t *mapped_node_list,
|
|
orte_app_context_t *app,
|
|
opal_list_t *master_node_list,
|
|
orte_std_cntr_t *total_num_slots);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_claim_slot(orte_job_t *jdata,
|
|
orte_node_t *current_node,
|
|
int32_t stride,
|
|
orte_std_cntr_t app_idx,
|
|
opal_list_t *nodes,
|
|
bool oversubscribe,
|
|
bool remove_from_list,
|
|
orte_proc_t **returnproc);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_compute_vpids(orte_job_t *jdata);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_compute_local_ranks(orte_job_t *jdata);
|
|
|
|
ORTE_DECLSPEC void orte_rmaps_base_update_local_ranks(orte_job_t *jdata, orte_node_t *oldnode,
|
|
orte_node_t *newnode, orte_proc_t *newproc);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_rearrange_map(orte_app_context_t *app, orte_job_map_t *map, opal_list_t *procs);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_define_daemons(orte_job_t *jdata);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_setup_virtual_machine(orte_job_t *jdata);
|
|
|
|
ORTE_DECLSPEC opal_list_item_t* orte_rmaps_base_get_starting_point(opal_list_t *node_list, orte_job_t *jdata);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_map_byslot(orte_job_t *jdata, orte_app_context_t *app,
|
|
opal_list_t *node_list, orte_vpid_t num_procs,
|
|
opal_list_item_t *cur_node_item);
|
|
|
|
ORTE_DECLSPEC int orte_rmaps_base_map_bynode(orte_job_t *jdata, orte_app_context_t *app,
|
|
opal_list_t *node_list, orte_vpid_t num_procs,
|
|
opal_list_item_t *cur_node_item);
|
|
|
|
|
|
END_C_DECLS
|
|
|
|
#endif
|