monkey patch

log_inductor_output.py

from torch._logging._internal import trace_structured # noqa: E402
import torch._inductor.codecache # noqa: E402
import torch._inductor.graph # noqa: E402
def _patched_trace_structured(name, *args, **kwargs):
    if name == "inductor_output_code":
        match args, kwargs:
            case (metadata_fn, *_), _:
                filename = metadata_fn().get("filename", "Unknown")
            case _, {"metadata_fn": metadata_fn}:
                filename = metadata_fn().get("filename", "Unknown")
            case _:
                filename = "Unknown"
        print(f"inductor_output_code: {filename}")
    trace_structured(name, *args, **kwargs)
torch._inductor.codecache.trace_structured = _patched_trace_structured
torch._inductor.graph.trace_structured = _patched_trace_structured