openpmix-src 0.1.0-rc.1

Vendored build of OpenPMIx, developed for use by the Lamellar runtime.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149

/*
 * Copyright (c) 2015-2020 Intel, Inc.  All rights reserved.
 *
 * Copyright (c) 2021-2025 Nanook Consulting  All rights reserved.
 * $COPYRIGHT$
 *
 * Additional copyrights may follow
 *
 * $HEADER$
 */

#ifndef PMIX_GDS_HASH_H
#define PMIX_GDS_HASH_H

#include "src/include/pmix_config.h"

#include "src/class/pmix_list.h"
#include "src/include/pmix_globals.h"
#include "src/util/pmix_argv.h"
#include "src/util/pmix_error.h"
#include "src/util/pmix_hash.h"
#include "src/util/pmix_name_fns.h"
#include "src/util/pmix_output.h"

#include "src/mca/gds/gds.h"

BEGIN_C_DECLS

typedef struct {
    pmix_gds_base_component_t super;
    pmix_list_t mysessions;
    pmix_list_t myjobs;
} pmix_gds_hash_component_t;

/* the component must be visible data for the linker to find it */
PMIX_EXPORT extern pmix_gds_hash_component_t pmix_mca_gds_hash_component;
extern pmix_gds_base_module_t pmix_hash_module;

/* Define a bitmask to track what information may not have
 * been provided but is computable from other info */
#define PMIX_HASH_PROC_DATA 0x00000001
#define PMIX_HASH_JOB_SIZE  0x00000002
#define PMIX_HASH_MAX_PROCS 0x00000004
#define PMIX_HASH_NUM_NODES 0x00000008
#define PMIX_HASH_PROC_MAP  0x00000010
#define PMIX_HASH_NODE_MAP  0x00000020

/* struct definitions */
typedef struct {
    pmix_list_item_t super;
    uint32_t session;
    pmix_list_t sessioninfo;
    pmix_list_t nodeinfo;
} pmix_session_t;
PMIX_CLASS_DECLARATION(pmix_session_t);

typedef struct {
    pmix_list_item_t super;
    char *ns;
    pmix_namespace_t *nptr;
    pmix_hash_table_t internal;
    pmix_hash_table_t remote;
    pmix_hash_table_t local;
    bool gdata_added;
    pmix_list_t jobinfo;
    pmix_list_t apps;
    pmix_list_t nodeinfo;
    pmix_session_t *session;
} pmix_job_t;
PMIX_CLASS_DECLARATION(pmix_job_t);

typedef struct {
    pmix_list_item_t super;
    uint32_t appnum;
    pmix_list_t appinfo;
    pmix_list_t nodeinfo;
    pmix_job_t *job;
} pmix_apptrkr_t;
PMIX_CLASS_DECLARATION(pmix_apptrkr_t);

typedef struct {
    pmix_list_item_t super;
    uint32_t nodeid;
    char *hostname;
    char **aliases;
    pmix_list_t info;
} pmix_nodeinfo_t;
PMIX_CLASS_DECLARATION(pmix_nodeinfo_t);

extern pmix_status_t pmix_gds_hash_process_node_array(pmix_value_t *val, pmix_list_t *tgt);

extern pmix_status_t pmix_gds_hash_process_app_array(pmix_value_t *val, pmix_job_t *trk);

extern pmix_status_t pmix_gds_hash_process_job_array(pmix_info_t *info, pmix_job_t *trk,
                                                     uint32_t *flags, char ***procs, char ***nodes);

extern pmix_status_t pmix_gds_hash_process_session_array(pmix_value_t *val, pmix_job_t *trk);

extern pmix_job_t *pmix_gds_hash_get_tracker(const pmix_nspace_t nspace, bool create);

extern pmix_session_t* pmix_gds_hash_check_session(pmix_job_t *trk,
                                                   uint32_t sid,
                                                   bool create);
extern pmix_status_t pmix_gds_hash_xfer_sessioninfo(pmix_peer_t *peer,
                                                    pmix_session_t *sptr,
                                                    const char *key,
                                                    pmix_list_t *kvs);

extern bool pmix_gds_hash_check_hostname(char *h1, char *h2);

extern bool pmix_gds_hash_check_node(pmix_nodeinfo_t *n1, pmix_nodeinfo_t *n2);

extern pmix_nodeinfo_t* pmix_gds_hash_check_nodename(pmix_list_t *nodes, char *hostname);

extern pmix_status_t pmix_gds_hash_store_map(pmix_job_t *trk, char **nodes, char **ppn,
                                             uint32_t flags);

extern pmix_status_t pmix_gds_hash_fetch(struct pmix_peer_t *peer,
                                         const pmix_proc_t *proc, pmix_scope_t scope, bool copy,
                                         const char *key, pmix_info_t qualifiers[], size_t nqual,
                                         pmix_list_t *kvs);

extern pmix_status_t pmix_gds_hash_fetch_sessioninfo(pmix_peer_t *peer,
                                                     const char *key,
                                                     pmix_job_t *trk,
                                                     pmix_info_t *info, size_t ninfo,
                                                     pmix_list_t *kvs);

extern pmix_status_t pmix_gds_hash_fetch_nodeinfo(pmix_peer_t *peer,
                                                  const char *key,
                                                  pmix_list_t *tgt, pmix_info_t *info, size_t ninfo,
                                                  pmix_list_t *kvs);

extern pmix_status_t pmix_gds_hash_fetch_appinfo(pmix_peer_t *peer,
                                                 const char *key,  pmix_list_t *tgt,
                                                 pmix_info_t *info, size_t ninfo, pmix_list_t *kvs);

extern pmix_status_t pmix_gds_hash_store(const pmix_proc_t *proc, pmix_scope_t scope,
                                         pmix_kval_t *kv);

extern pmix_status_t pmix_gds_hash_store_qualified(pmix_hash_table_t *ht,
                                                   pmix_rank_t rank,
                                                   pmix_value_t *value);

extern pmix_status_t pmix_gds_hash_fetch_arrays(struct pmix_peer_t *pr, pmix_buffer_t *reply);

END_C_DECLS

#endif