| #!/usr/bin/env python3 | |
| """Run a command and report the peak RSS (GiB) and wall time of the process tree on stderr. | |
| usage: measure_rss.py <cmd> [args...] | |
| Exit code is the child's exit code. | |
| """ | |
| import resource | |
| import subprocess | |
| import sys | |
| import time | |
| t0 = time.time() | |
| rc = subprocess.call(sys.argv[1:]) | |
| peak_gib = resource.getrusage(resource.RUSAGE_CHILDREN).ru_maxrss / 1024 / 1024 | |
| print(f"EXIT={rc} PEAK_RSS={peak_gib:.2f} GiB WALL={time.time() - t0:.1f}s", file=sys.stderr) | |
| sys.exit(rc) | |