torchref.utils.timing module

CLI timing utility — register once at the start of main().

torchref.utils.timing.register_timing()

Register an atexit handler that prints wall and CPU time on exit.

Call at the top of main() in CLI scripts:

from torchref.utils.timing import register_timing

def main():
    register_timing()
    ...