More consolidation of MPI initialization
[alexxy/gromacs.git] / src / gromacs / legacyheaders / network.h
1 /*
2  *
3  *                This source code is part of
4  *
5  *                 G   R   O   M   A   C   S
6  *
7  *          GROningen MAchine for Chemical Simulations
8  *
9  *                        VERSION 3.2.0
10  * Written by David van der Spoel, Erik Lindahl, Berk Hess, and others.
11  * Copyright (c) 1991-2000, University of Groningen, The Netherlands.
12  * Copyright (c) 2001-2004, The GROMACS development team,
13  * check out http://www.gromacs.org for more information.
14
15  * This program is free software; you can redistribute it and/or
16  * modify it under the terms of the GNU General Public License
17  * as published by the Free Software Foundation; either version 2
18  * of the License, or (at your option) any later version.
19  *
20  * If you want to redistribute modifications, please consider that
21  * scientific software is very special. Version control is crucial -
22  * bugs must be traceable. We will be happy to consider code for
23  * inclusion in the official distribution, but derived work must not
24  * be called official GROMACS. Details are found in the README & COPYING
25  * files - if they are missing, get the official version at www.gromacs.org.
26  *
27  * To help us fund GROMACS development, we humbly ask that you cite
28  * the papers on the package - you can find them in the top README file.
29  *
30  * For more info, check our website at http://www.gromacs.org
31  *
32  * And Hey:
33  * Gromacs Runs On Most of All Computer Systems
34  */
35
36 #ifndef _network_h
37 #define _network_h
38
39
40 /*
41  * This module defines the interface of the actual communication routines.
42  */
43
44 #include <stdio.h>
45
46 #include "types/simple.h"
47 #include "types/commrec.h"
48 #include "typedefs.h"
49 #include "main.h"
50 #include "gmx_fatal.h"
51
52 #ifdef __cplusplus
53 extern "C" {
54 #endif
55
56 t_commrec *init_commrec(void);
57 /* Allocate, initialize and return the commrec. */
58
59 t_commrec *init_par_threads(const t_commrec *cro);
60 /* Initialize communication records for thread-parallel simulations.
61    Must be called on all threads before any communication takes place by
62    the individual threads. Copies the original commrec to
63    thread-local versions (a small memory leak results because we don't
64    deallocate the old shared version).  */
65
66 void gmx_fill_commrec_from_mpi(t_commrec *cr);
67 /* Continues t_commrec construction */
68
69 int gmx_node_num(void);
70 /* return the number of nodes in the ring */
71
72 int gmx_node_rank(void);
73 /* return the rank of the node */
74
75 int gmx_hostname_num(void);
76 /* If the first part of the hostname (up to the first dot) ends with a number, returns this number.
77    If the first part of the hostname does not ends in a number (0-9 characters), returns 0.
78  */
79
80 void gmx_setup_nodecomm(FILE *fplog, t_commrec *cr);
81 /* Sets up fast global communication for clusters with multi-core nodes */
82
83 void gmx_init_intranode_counters(t_commrec *cr);
84 /* Initializes intra-physical-node MPI process/thread counts and ID. */
85
86 gmx_bool gmx_mpi_initialized(void);
87 /* return TRUE when MPI_Init has been called.
88  * return FALSE when MPI_Init has not been called OR
89  * when GROMACS was compiled without MPI support.
90  */
91
92 void gmx_barrier(const t_commrec *cr);
93 /* Wait till all processes in cr->mpi_comm_mygroup have reached the barrier */
94
95 void gmx_bcast(int nbytes, void *b, const t_commrec *cr);
96 /* Broadcast nbytes bytes from the master to cr->mpi_comm_mygroup */
97
98 void gmx_bcast_sim(int nbytes, void *b, const t_commrec *cr);
99 /* Broadcast nbytes bytes from the sim master to cr->mpi_comm_mysim */
100
101 void gmx_sumi(int nr, int r[], const t_commrec *cr);
102 /* Calculate the global sum of an array of ints */
103
104 void gmx_sumli(int nr, gmx_large_int_t r[], const t_commrec *cr);
105 /* Calculate the global sum of an array of large ints */
106
107 void gmx_sumf(int nr, float r[], const t_commrec *cr);
108 /* Calculate the global sum of an array of floats */
109
110 void gmx_sumd(int nr, double r[], const t_commrec *cr);
111 /* Calculate the global sum of an array of doubles */
112
113 void gmx_sumf_comm(int nr, float r[], MPI_Comm mpi_comm);
114 /* Calculate the global sum of an array of floats */
115
116 void gmx_sumd_comm(int nr, double r[], MPI_Comm mpi_comm);
117 /* Calculate the global sum of an array of doubles */
118
119 void gmx_sumi_sim(int nr, int r[], const gmx_multisim_t *ms);
120 /* Calculate the sum over the simulations of an array of ints */
121
122 void gmx_sumli_sim(int nr, gmx_large_int_t r[], const gmx_multisim_t *ms);
123 /* Calculate the sum over the simulations of an array of large ints */
124
125 void gmx_sumf_sim(int nr, float r[], const gmx_multisim_t *ms);
126 /* Calculate the sum over the simulations of an array of floats */
127
128 void gmx_sumd_sim(int nr, double r[], const gmx_multisim_t *ms);
129 /* Calculate the sum over the simulations of an array of doubles */
130
131 void gmx_abort(int nodeid, int nnodes, int errorno);
132 /* Abort the parallel run */
133
134 #ifdef GMX_DOUBLE
135 #define gmx_sum_comm  gmx_sumd_comm
136 #define gmx_sum       gmx_sumd
137 #define gmx_sum_sim   gmx_sumd_sim
138 #else
139 #define gmx_sum_comm  gmx_sumf_comm
140 #define gmx_sum       gmx_sumf
141 #define gmx_sum_sim   gmx_sumf_sim
142 #endif
143
144 #ifdef DEBUG_GMX
145 #define debug_gmx() do { FILE *fp = debug ? debug : stderr; \
146                          if (bDebugMode()) { fprintf(fp, "NODEID=%d, %s  %d\n", gmx_mpi_initialized() ? gmx_node_rank() : -1, __FILE__, __LINE__); } fflush(fp); } while (0)
147 #else
148 #define debug_gmx()
149 #endif
150
151 #ifdef __cplusplus
152 }
153 #endif
154
155
156 #endif  /* _network_h */