File nsys_wrapper.py

benchmarking/nsys_wrapper.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	"""Per-rank nsys wrapper for torchrun.
2
3	torchrun launches this script as each worker. It reads LOCAL_RANK from
4	the environment, wraps the actual speed_test.py invocation with its own

nothing calls this directly

mainFunction · 0.70

no test coverage detected