From 760d355e838542dda91234c5644b711012c5127e Mon Sep 17 00:00:00 2001
From: robjmcgibbon <robjmcgibbon@gmail.com>
Date: Mon, 8 Dec 2025 12:07:40 +0000
Subject: [PATCH] Add timing to matching script

---
 misc/match_group_membership.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/misc/match_group_membership.py b/misc/match_group_membership.py
index 7424a9ba..081078b1 100644
--- a/misc/match_group_membership.py
+++ b/misc/match_group_membership.py
@@ -23,6 +23,7 @@
 """
 
 import argparse
+import datetime
 import os
 
 from mpi4py import MPI
@@ -319,6 +320,8 @@ def mpi_print(string, comm_rank):
 
 if __name__ == "__main__":
 
+    start_time = datetime.datetime.now()
+
     parser = argparse.ArgumentParser(
         description=(
             "Script to match halos across runs by comparing particles"
@@ -503,4 +506,6 @@ def mpi_print(string, comm_rank):
         phdf5.collective_write(file, "MatchCount2to1", match_count_21, comm=comm)
         phdf5.collective_write(file, "Consistent2to1", consistent_21, comm=comm)
 
+    comm.barrier()
+    mpi_print(f"Runtime: {datetime.datetime.now() - start_time}", comm_rank)
     mpi_print("Done!", comm_rank)