https://github.com/pierre-guillou/pdiags_bench
Tip revision: 8a2de479abf7bcbf2002b4f8f620c635217cc088 authored by Julien J Tierny on 30 January 2023, 16:30:54 UTC
added reference to the arxiv repo
added reference to the arxiv repo
Tip revision: 8a2de47
subprocess_wrapper.py
import argparse
import resource
import subprocess
import sys
import time
def main(cmd):
beg = time.time()
subprocess.run(cmd, check=True)
end = time.time()
res = resource.getrusage(resource.RUSAGE_CHILDREN)
print(f"Elapsed Time (s): {end - beg}", file=sys.stderr)
print(f"Peak Memory (kB): {res.ru_maxrss}", file=sys.stderr)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="GNU Time replacement")
parser.add_argument("cmd", nargs="+")
args = parser.parse_args()
main(args.cmd)