mirror of
https://github.com/NGSolve/netgen.git
synced 2024-12-27 14:20:34 +05:00
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
functions = [
|
|
("int", "MPI_Init", "int*", "char***"),
|
|
("int", "MPI_Initialized", "int*"),
|
|
("int", "MPI_Finalize"),
|
|
("int", "MPI_Barrier", "MPI_Comm"),
|
|
("int", "MPI_Comm_free", "MPI_Comm*"),
|
|
("int", "MPI_Comm_rank", "MPI_Comm", "int*"),
|
|
("int", "MPI_Comm_size", "MPI_Comm", "int*"),
|
|
("int", "MPI_Type_commit", "MPI_Datatype*"),
|
|
("int", "MPI_Waitall", "int", "MPI_Request*", "MPI_Status*"),
|
|
("int", "MPI_Waitany", "int", "MPI_Request*", "int*", "MPI_Status*"),
|
|
("int", "MPI_Wait", "MPI_Request*", "MPI_Status*"),
|
|
("int", "MPI_Send", "void*", "int", "MPI_Datatype", "int", "int", "MPI_Comm"),
|
|
("int", "MPI_Probe", "int", "int", "MPI_Comm", "MPI_Status*"),
|
|
("int", "MPI_Recv", "void*", "int", "MPI_Datatype", "int", "int", "MPI_Comm", "MPI_Status*"),
|
|
("int", "MPI_Get_count", "MPI_Status*", "MPI_Datatype", "int*"),
|
|
("int", "MPI_Bcast", "void*", "int", "MPI_Datatype", "int", "MPI_Comm"),
|
|
("int", "MPI_Comm_group", "MPI_Comm", "MPI_Group*"),
|
|
("int", "MPI_Group_incl", "MPI_Group", "int", "int*", "MPI_Group*"),
|
|
("int", "MPI_Comm_create_group", "MPI_Comm", "MPI_Group", "int", "MPI_Comm*"),
|
|
("int", "MPI_Get_processor_name", "char*", "int*"),
|
|
]
|
|
|
|
constants = [
|
|
("MPI_Comm", "MPI_COMM_WORLD"),
|
|
("MPI_Status*", "MPI_STATUS_IGNORE"),
|
|
("MPI_Status*", "MPI_STATUSES_IGNORE"),
|
|
("MPI_Datatype", "MPI_INT"),
|
|
("MPI_Datatype", "MPI_SHORT"),
|
|
("MPI_Datatype", "MPI_CHAR"),
|
|
("MPI_Datatype", "MPI_UINT64_T"),
|
|
("MPI_Datatype", "MPI_DOUBLE"),
|
|
("MPI_Datatype", "MPI_C_BOOL"),
|
|
("void*", "MPI_IN_PLACE"),
|
|
("int", "MPI_MAX_PROCESSOR_NAME"),
|
|
]
|
|
|
|
def get_args(f):
|
|
return ["NG_"+a if a.startswith("MPI_") else a for a in f[2:]]
|
|
|
|
def generate_declarations():
|
|
code = ""
|
|
for f in functions:
|
|
ret = f[0]
|
|
name = f[1]
|
|
args = ", ".join(get_args(f))
|
|
code += f"NGCORE_API extern {ret} (*NG_{name})({args});\n"
|
|
|
|
for typ, name in constants:
|
|
if typ.startswith("MPI_"):
|
|
typ = "NG_" + typ
|
|
code += f"NGCORE_API extern {typ} NG_{name};\n"
|
|
|
|
with open("ng_mpi_generated_declarations.hpp", "w") as f:
|
|
f.write(code)
|
|
|
|
def generate_dummy_init():
|
|
code = ""
|
|
for f in functions:
|
|
ret = f[0]
|
|
name = f[1]
|
|
args = ", ".join(get_args(f))
|
|
code += f"decltype(NG_{name}) NG_{name} = []({args})->{ret} {{ throw no_mpi(); }};\n"
|
|
|
|
for typ, name in constants:
|
|
if typ.startswith("MPI_"):
|
|
typ = "NG_" + typ
|
|
code += f"{typ} NG_{name} = 0;\n"
|
|
|
|
with open("ng_mpi_generated_dummy_init.hpp", "w") as f:
|
|
f.write(code)
|
|
|
|
def generate_init():
|
|
code = ""
|
|
for f in functions:
|
|
ret = f[0]
|
|
name = f[1]
|
|
args = get_args(f)
|
|
in_args =''
|
|
call_args = ''
|
|
for i, a in enumerate(args):
|
|
if i > 0:
|
|
in_args += ', '
|
|
call_args += ', '
|
|
in_args += a + f" arg{i}"
|
|
call_args += f" ng2mpi(arg{i})"
|
|
code += f"NG_{name} = []({in_args})->{ret} {{ return {name}({call_args}); }};\n"
|
|
|
|
for _, name in constants:
|
|
code += f"NG_{name} = mpi2ng({name});\n"
|
|
|
|
with open("ng_mpi_generated_init.hpp", "w") as f:
|
|
f.write(code)
|
|
|
|
if __name__ == "__main__":
|
|
generate_declarations()
|
|
generate_dummy_init()
|
|
generate_init()
|