Skip to content
Snippets Groups Projects

P2P

Open
Jammer, Timrequested to merge
p2p into main
2 files
+ 24
22
Compare changes
  • Side-by-side
  • Inline

Files

@@ -112,9 +112,9 @@ def get_send_recv_template(send_func: str = "mpi_isend", recv_func: str | typing
if recv_func in probe_pairs:
if recv_func in [["mpi_improbe", "mpi_mrecv"],
["mpi_improbe", "mpi_imrecv"]]:
tm.insert_instruction(Instruction("while (!" + flag_name + "){", rank=0), before_instruction=r)
tm.insert_instruction(Instruction("while (!" + flag_name + "){", rank=0,identifier="PROBELOOP"), before_instruction=r)
# insertion before the improbe call
tm.register_instruction("}", rank_to_execute=0) # end while
tm.register_instruction("}", rank_to_execute=0,identifier="PROBELOOP") # end while
# the matched recv
tm.register_instruction(CorrectMPICallFactory.get(recv_func[1]), rank_to_execute=0)
Loading