fix issues

Author: xadupre

_doc/conf.py

@@ -152,7 +152,6 @@
     "within_subsection_order": "ExampleTitleSortKey",
     # errors
     "abort_on_example_error": True,
-    "expected_failing_examples": ["examples/plot_export_locate_issue.py"],
     # recommendation
     "recommender": {"enable": True, "n_examples": 3, "min_df": 3, "max_df": 0.9},
     # ignore capture for matplotib axes

_doc/examples/plot_export_locate_issue.py

@@ -5,6 +5,14 @@
 Find and fix an export issue due to dynamic shapes
 ==================================================
 
+LLMs must be exported with dynamic shapes and it is common that
+a static dimension turns into a static ones. The error message from
+:epkg:`pytorch` tells the user to define ``TORCH_LOGS="+dynamic"``
+but it shows a very long list of messages where we need
+to find the string ``range_refined_to_singleton`` and that
+does not really indicates where it comes from. The example
+shows how to tweak pytorch to get that information until
+it gets better.
 
 A model with an export issue
 ============================
@@ -15,6 +23,7 @@
 it is difficult to find deep inside a big model.
 """
 
+import traceback
 import torch
 from onnx_diagnostic.torch_export_patches import bypass_export_some_errors
 
@@ -69,10 +78,17 @@ def forward(self, x: torch.Tensor, ys: list[torch.Tensor]):
 # Stop when a dynamic dimension turns static
 # ==========================================
 #
-#
+# We use :func:`bypass_export_some_errors
+# <onnx_diagnostic.torch_export_patches.bypass_export_some_errors>`
+# to replace torch implementation by a new one raising the exception
+# mentioned in previous section.
 
 with bypass_export_some_errors(stop_if_static=True, verbose=1):
-    torch.export.export(model, inputs, dynamic_shapes=dyn_shapes)
+    try:
+        torch.export.export(model, inputs, dynamic_shapes=dyn_shapes)
+    except AssertionError:
+        print("-- It failed as excepted. Let's print the stack trace.")
+        print(traceback.format_exc())
 
 # The stack trace is quite long but the first line referring to this example
 # is the following one. It points out the line turing a dynamic dimension into

onnx_diagnostic/torch_export_patches/onnx_export_errors.py

@@ -210,7 +210,7 @@ def bypass_export_some_errors(
         the environment variable ``SKIP_SOLVE_CONSTRAINTS=0``
         can be put to stop at that stage.
     :param stop_if_static: see example :ref:`l-plot-export-locale-issue`,
-        to stop the export as soon as an issue is detected with dyanmic shapes
+        to stop the export as soon as an issue is detected with dynamic shapes
         and show a stack trace indicating the exact location of the issue
     :param patch: if False, disable all patches except the registration of
         serialization function
@@ -361,7 +361,7 @@ def bypass_export_some_errors(
         if stop_if_static:
             if verbose:
                 print(
-                    "[bypass_export_some_errors] assert when a dynamic dimnension turns static"
+                    "[bypass_export_some_errors] assert when a dynamic dimension turns static"
                 )
 
             from torch.fx.experimental.symbolic_shapes import ShapeEnv