doc

Author: xadupre

README.rst

@@ -50,6 +50,8 @@ Enlightening Examples
 
 * `Use DYNAMIC or AUTO when exporting if dynamic shapes has constraints
   <https://sdpython.github.io/doc/onnx-diagnostic/dev/auto_examples/plot_export_with_dynamic_shapes_auto.html>`_
+* `Find and fix an export issue due to dynamic shapes
+  <https://sdpython.github.io/doc/onnx-diagnostic/dev/auto_examples/plot_export_locate_issue.html>`_
 * `Export with DynamicCache and dynamic shapes
   <https://sdpython.github.io/doc/onnx-diagnostic/dev/auto_examples/plot_export_with_dynamic_cache.html>`_
 * `Steel method forward to guess the dynamic shapes (with Tiny-LLM)

_doc/examples/plot_export_tiny_llm.py

@@ -44,10 +44,11 @@
 
 def _forward_(*args, _f=None, **kwargs):
     assert _f is not None
-    if not torch.compiler.is_exporting():
+    if not hasattr(torch.compiler, "is_exporting") or not torch.compiler.is_exporting():
+        # torch.compiler.is_exporting requires torch>=2.7
         print("<-", string_type((args, kwargs), with_shape=True, with_min_max=True))
     res = _f(*args, **kwargs)
-    if not torch.compiler.is_exporting():
+    if not hasattr(torch.compiler, "is_exporting") or not torch.compiler.is_exporting():
         print("->", string_type((args, kwargs), with_shape=True, with_min_max=True))
     return res
 

_doc/index.rst

@@ -61,6 +61,7 @@ Enlightening Examples
 * :ref:`l-plot-export-cond`
 * :ref:`l-plot-sxport-with-dynamio-shapes-auto`
 * :ref:`l-plot-export-with-dynamic-shape`
+* :ref:`l-plot-export-locale-issue`
 * :ref:`l-plot-tiny-llm-export`
 * :ref:`l-plot-tiny-llm-export-patched`
 
@@ -162,5 +163,7 @@ Size of the package:
 Older versions
 ++++++++++++++
 
+* `0.2.2 <../v0.2.2/index.html>`_
+* `0.2.1 <../v0.2.1/index.html>`_
 * `0.2.0 <../v0.2.0/index.html>`_
 * `0.1.0 <../v0.1.0/index.html>`_

onnx_diagnostic/__init__.py

@@ -3,14 +3,5 @@
 Functions, classes to dig into a model when this one is right, slow, wrong...
 """
 
-__version__ = "0.3.0"
+__version__ = "0.2.2"
 __author__ = "Xavier Dupré"
-
-
-def reset_torch_transformers(gallery_conf, fname):
-    "Resets torch dynamo for :epkg:`sphinx-gallery`."
-    import matplotlib.pyplot as plt
-    import torch
-
-    plt.style.use("ggplot")
-    torch._dynamo.reset()

onnx_diagnostic/torch_test_helper.py

@@ -13,6 +13,10 @@ def is_torchdynamo_exporting() -> bool:
     """Tells if torch is exporting a model."""
     import torch
 
+    if not hasattr(torch.compiler, "is_exporting"):
+        # torch.compiler.is_exporting requires torch>=2.7
+        return False
+
     try:
         return torch.compiler.is_exporting()
     except Exception: