Spaces:
Running
Running
| # SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. | |
| # SPDX-License-Identifier: OpenMDW-1.1 | |
| # Runtime load tracer. | |
| # | |
| # Auto-imported by every Python process started with PYTHONPATH=. | |
| # When LOAD_TRACE_DIR is set, registers an atexit hook that walks | |
| # sys.modules at shutdown and writes the file paths (filtered to those | |
| # under LOAD_TRACE_ROOT) into {LOAD_TRACE_DIR}/{LOAD_TRACE_TAG}_pid{PID}.txt. | |
| # | |
| # Used to inventory which released files are actually touched by each | |
| # end-to-end smoke. Union the per-experiment traces, diff against the full | |
| # .py list, and the residual is dead code (relative to that smoke set). | |
| import atexit | |
| import os | |
| import sys | |
| _DIR = os.environ.get("LOAD_TRACE_DIR", "") | |
| if _DIR: | |
| _TAG = os.environ.get("LOAD_TRACE_TAG", "default") | |
| _ROOT = os.path.realpath(os.environ.get("LOAD_TRACE_ROOT", os.getcwd())) | |
| os.makedirs(_DIR, exist_ok=True) | |
| def _dump(): | |
| seen = set() | |
| for mod in list(sys.modules.values()): | |
| f = getattr(mod, "__file__", None) | |
| if not f: | |
| continue | |
| try: | |
| rp = os.path.realpath(f) | |
| except OSError: | |
| continue | |
| if rp.startswith(_ROOT): | |
| seen.add(rp) | |
| path = os.path.join(_DIR, f"{_TAG}_pid{os.getpid()}.txt") | |
| try: | |
| with open(path, "w") as h: | |
| for p in sorted(seen): | |
| h.write(p + "\n") | |
| except OSError: | |
| pass | |
| atexit.register(_dump) | |