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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
##
## Copyright (C) by Argonne National Laboratory
## See COPYRIGHT in top-level directory
##
from local_python import MPI_API_Global as G
from local_python.mpi_api import *
from local_python.binding_common import *
from local_python.binding_f08 import *
from local_python import RE
import os
def main():
# currently support -no-real128, -no-mpiio, -fint-size, -aint-size, -count-size, -cint-size
G.parse_cmdline()
binding_dir = G.get_srcdir_path("src/binding")
f08_dir = "src/binding/fortran/use_mpi_f08"
G.check_write_path("%s/wrappers_f/" % f08_dir)
G.check_write_path("%s/wrappers_c/" % f08_dir)
func_list = load_C_func_list(binding_dir, True) # suppress noise
if "no-mpiio" in G.opts:
# a few MPI_File_xxx functions are already in (MPI_File_xxx_errhandler)
func_list = [f for f in func_list if not f['name'].startswith('MPI_File_')]
else:
# FIXME: until romio interface is generated
func_list.extend(get_mpiio_func_list())
func_list.append(G.FUNCS['mpi_f_sync_reg'])
# preprocess
get_real_POLY_kinds()
for func in func_list:
check_func_directives(func)
if '_skip_fortran' in func:
continue
if re.match(r'mpi_op_create|mpi_register_datarep', func['name'], re.IGNORECASE):
func['_need_large'] = True
# need separate interface, e.g. MPI_Op_create_c
func['_need_large_separate'] = True
elif function_has_real_POLY_parameters(func):
func['_need_large'] = True
else:
func['_need_large'] = False
process_func_parameters(func)
func_list = [f for f in func_list if '_skip_fortran' not in f]
# f08_cdesc.c
G.out = []
G.decls = []
for func in func_list:
if need_cdesc(func):
G.out.append("")
dump_f08_wrappers_c(func, False)
if func['_need_large']:
G.out.append("")
dump_f08_wrappers_c(func, True)
f = "%s/wrappers_c/f08_cdesc.c" % f08_dir
dump_cdesc_c(f, G.out)
f = "%s/wrappers_c/cdesc_proto.h" % f08_dir
dump_cdesc_proto_h(f, G.decls)
# f08ts.f90
G.out = []
for func in func_list:
dump_f08_wrappers_f(func, False)
if func['_need_large']:
dump_f08_wrappers_f(func, True)
f = "%s/wrappers_f/f08ts.f90" % f08_dir
dump_f90_file(f, G.out)
do_profiling = True
if do_profiling:
temp_out = []
for l in G.out:
temp_out.append(re.sub(r'(subroutine|function)\s+(MPIX?)_', r'\1 P\2R_', l, flags=re.IGNORECASE))
f = "%s/wrappers_f/pf08ts.f90" % f08_dir
dump_f90_file(f, temp_out)
temp_out = None
# mpi_c_interface_{cdesc,nobuf}.f90
G.out = []
dump_interface_module_open("mpi_c_interface_cdesc")
for func in func_list:
if need_cdesc(func):
dump_mpi_c_interface_cdesc(func, False)
if func['_need_large']:
dump_mpi_c_interface_cdesc(func, True)
dump_interface_module_close("mpi_c_interface_cdesc")
f = "%s/mpi_c_interface_cdesc.f90" % f08_dir
dump_f90_file(f, G.out)
G.out = []
dump_interface_module_open("mpi_c_interface_nobuf")
for func in func_list:
if not need_cdesc(func):
dump_mpi_c_interface_nobuf(func, False)
if func['_need_large']:
dump_mpi_c_interface_nobuf(func, True)
dump_interface_module_close("mpi_c_interface_nobuf")
f = "%s/mpi_c_interface_nobuf.f90" % f08_dir
dump_f90_file(f, G.out)
# mpi_f08.f90 and pmpi_f08.f90
G.out = []
dump_F_module_open("mpi_f08")
G.out.append("USE,intrinsic :: iso_c_binding, ONLY: c_ptr")
G.out.append("USE :: pmpi_f08")
G.out.append("USE :: mpi_f08_types")
G.out.append("USE :: mpi_f08_compile_constants")
G.out.append("USE :: mpi_f08_link_constants")
G.out.append("USE :: mpi_f08_callbacks")
G.out.append("")
G.out.append("IMPLICIT NONE")
for func in func_list:
G.out.append("")
func_name = get_function_name(func, False)
G.out.append("INTERFACE %s" % func_name)
G.out.append("INDENT")
dump_mpi_f08(func, False)
if func['_need_large'] and '_need_large_separate' not in func:
G.out.append("")
dump_mpi_f08(func, True)
G.out.append("DEDENT")
G.out.append("END INTERFACE %s" % func_name)
if '_need_large_separate' in func:
G.out.append("")
func_name = get_function_name(func, False) + "_c"
G.out.append("INTERFACE %s" % func_name)
G.out.append("INDENT")
dump_mpi_f08(func, True)
G.out.append("DEDENT")
G.out.append("END INTERFACE %s" % func_name)
G.out.append("")
dump_F_module_close("mpi_f08")
f = "%s/mpi_f08.f90" % f08_dir
dump_f90_file(f, G.out)
if do_profiling:
temp_out = []
for l in G.out:
if l == "USE :: pmpi_f08":
pass
elif RE.match(r'((?:\s*end\s+)?module)\s+(\w+)', l, re.IGNORECASE):
temp_out.append(RE.m.group(1) + ' p' + RE.m.group(2))
elif RE.match(r'((?:\s*end\s+)?interface)\s+(\w+)', l, re.IGNORECASE):
temp_out.append(RE.m.group(1) + ' P' + RE.m.group(2))
else:
temp_out.append(re.sub(r'(subroutine|function)\s+(MPIX?)_', r'\1 P\2R_', l, flags=re.IGNORECASE))
f = "%s/pmpi_f08.f90" % f08_dir
dump_f90_file(f, temp_out)
temp_out = None
# mpi_f08_types.f90
G.out = []
dump_mpi_f08_types()
f = "%s/mpi_f08_types.f90" % f08_dir
dump_f90_file(f, G.out)
# .in files has to be generated in the source tree
if G.is_autogen():
# mpi_f08_compile_constants.f90.in
G.mpih_defines = {}
load_mpi_h_in("src/include/mpi.h.in")
load_mpi_h_in("src/mpi/romio/include/mpio.h.in")
f = "%s/mpi_f08_compile_constants.f90.in" % f08_dir
dump_compile_constants_f90(f)
# ---------------------------------------------------------
if __name__ == "__main__":
main()
|
