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
|
/*
* Copyright (c) 2013-2015 Mellanox Technologies, Inc.
* All rights reserved.
* Copyright (c) 2016-2019 Research Organization for Information Science
* and Technology (RIST). All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "oshmem_config.h"
#include <stdlib.h>
#include "opal/util/output.h"
#include "oshmem/constants.h"
#include "oshmem/include/shmem.h"
#include "oshmem/runtime/params.h"
#include "oshmem/runtime/runtime.h"
#include "oshmem/shmem/shmem_api_logger.h"
#if OSHMEM_PROFILING
#include "oshmem/include/pshmem.h"
#pragma weak shmem_init = pshmem_init
#pragma weak shmem_init_thread = pshmem_init_thread
#pragma weak start_pes = pstart_pes
#include "oshmem/shmem/c/profile-defines.h"
#endif
extern int oshmem_shmem_globalexit_status;
static inline void _shmem_init(int required, int *provided);
void shmem_init(void)
{
int provided;
/* spec says that npes are ignored for now */
_shmem_init(SHMEM_THREAD_SINGLE, &provided);
}
int shmem_init_thread(int requested, int *provided)
{
_shmem_init(requested, provided);
return 0;
}
void start_pes(int npes)
{
int provided;
/* spec says that npes are ignored for now */
_shmem_init(SHMEM_THREAD_SINGLE, &provided);
}
static void shmem_onexit(int exitcode, void *arg)
{
oshmem_shmem_globalexit_status = exitcode;
shmem_finalize();
}
static inline void _shmem_init(int required, int *provided)
{
int err = OSHMEM_SUCCESS;
if (oshmem_shmem_initialized) {
/*
* SPEC: If start_pes() is called multiple times, subsequent calls have no effect.
*/
return;
}
err = oshmem_shmem_init(0, NULL, required, provided);
if (OSHMEM_SUCCESS != err) {
/* since spec does not propagete error to user we can only abort */
SHMEM_API_ERROR("SHMEM failed to initialize - aborting");
oshmem_shmem_abort(-1);
}
#if HAVE_ON_EXIT
on_exit(shmem_onexit, NULL);
#endif
}
|
