[ONNX] Change stacklevel in warning message for export

Change to 3 so that the warning shows user callsite. (Where user calls torch.onnx.export)
2025-11-12 23:34:42 +08:00 · 2025-10-29 12:22:11 -07:00
1 changed files with 1 additions and 1 deletions
--- a/torch/onnx/_internal/exporter/_compat.py
+++ b/torch/onnx/_internal/exporter/_compat.py
@ -82,7 +82,7 @@ def export_compat(
                "different behavior during inference. "
                "Calling model.eval() before export is recommended.",
                UserWarning,
-                stacklevel=2,
+                stacklevel=3,
            )

    if isinstance(model, torch.export.ExportedProgram):