ras/base/base.h

0001 /*
0002  * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
0003  *                         University Research and Technology
0004  *                         Corporation.  All rights reserved.
0005  * Copyright (c) 2004-2005 The University of Tennessee and The University
0006  *                         of Tennessee Research Foundation.  All rights
0007  *                         reserved.
0008  * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
0009  *                         University of Stuttgart.  All rights reserved.
0010  * Copyright (c) 2004-2005 The Regents of the University of California.
0011  *                         All rights reserved.
0012  * Copyright (c) 2011-2013 Los Alamos National Security, LLC.  All rights
0013  *                         reserved.
0014  * Copyright (c) 2016-2020 Intel, Inc.  All rights reserved.
0015  * Copyright (c) 2020      Cisco Systems, Inc.  All rights reserved
0016  * Copyright (c) 2021-2025 Nanook Consulting  All rights reserved.
0017  * $COPYRIGHT$
0018  *
0019  * Additional copyrights may follow
0020  *
0021  * $HEADER$
0022  */
0023 /** @file:
0024  */
0025
0026 #ifndef PRTE_MCA_RAS_BASE_H
0027 #define PRTE_MCA_RAS_BASE_H
0028
0029 /*
0030  * includes
0031  */
0032 #include "prte_config.h"
0033 #include "src/mca/base/pmix_mca_base_framework.h"
0034
0035 #include "src/mca/ras/ras.h"
0036 #include "src/runtime/prte_globals.h"
0037 #include "src/util/pmix_printf.h"
0038 /*
0039  * Global functions for MCA overall collective open and close
0040  */
0041
0042 BEGIN_C_DECLS
0043
0044 /*
0045  * MCA Framework
0046  */
0047 PRTE_EXPORT extern pmix_mca_base_framework_t prte_ras_base_framework;
0048 /* select a component */
0049 PRTE_EXPORT int prte_ras_base_select(void);
0050
0051 /*
0052  * globals that might be needed
0053  */
0054 typedef struct prte_ras_base_t {
0055     bool first_pass_completed;
0056     bool allocation_read;
0057     prte_ras_base_module_t *active_module;
0058     int total_slots_alloc;
0059     int multiplier;
0060     bool launch_orted_on_hn;
0061     bool simulated;
0062 } prte_ras_base_t;
0063
0064 PRTE_EXPORT extern prte_ras_base_t prte_ras_base;
0065
0066 /**
0067  * Add the specified node definitions to the registry
0068  */
0069 PRTE_EXPORT int prte_ras_base_node_insert(pmix_list_t *, prte_job_t *);
0070
0071 PRTE_EXPORT void prte_ras_base_display_alloc(prte_job_t *jdata);
0072
0073 PRTE_EXPORT void prte_ras_base_display_cpus(prte_job_t *jdata, char *nodelist);
0074
0075 PRTE_EXPORT void prte_ras_base_allocate(int fd, short args, void *cbdata);
0076
0077 PRTE_EXPORT void prte_ras_base_modify(int fd, short args, void *cbdata);
0078
0079 PRTE_EXPORT int prte_ras_base_add_hosts(prte_job_t *jdata);
0080
0081 PRTE_EXPORT char *prte_ras_base_flag_string(prte_node_t *node);
0082
0083 END_C_DECLS
0084
0085 #endif