onnx-diagnostic 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

onnx_diagnostic/reference/torch_ops/sequence_ops.py

@@ -2,18 +2,18 @@ from typing import Optional
 import onnx
 import torch
 from ...helpers.torch_helper import onnx_dtype_to_torch_dtype
-from . import OpRun, OpRunSequence, OpRunTensor
+from . import OpRunKernel, OpRunSequence, OpRunTensor
 
 
-class OpRunOpSequence(OpRun):
+class OpRunOpSequence(OpRunKernel):
     "Ancestor for kernel using sequences."
 
 
 class ConcatFromSequence_11(OpRunOpSequence):
     "ConcatFromSequence"
 
-    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
-        super().__init__(node, version)
+    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None, verbose: int = 0):
+        super().__init__(node, version, verbose=verbose)
         axis = self.get_attribute_int(node, "axis", None)
         assert isinstance(axis, int), f"Unexpected value for attribute axis={axis!r}"
         self.axis = axis
@@ -39,8 +39,8 @@ class ConcatFromSequence_11(OpRunOpSequence):
 class SequenceEmpty_11(OpRunOpSequence):
     "SqeuenceEmpty"
 
-    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
-        super().__init__(node, version)
+    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None, verbose: int = 0):
+        super().__init__(node, version, verbose=verbose)
         self.dtype = onnx_dtype_to_torch_dtype(
             self.get_attribute_int(node, "dtype", onnx.TensorProto.FLOAT)  # type: ignore[arg-type]
         )

onnx_diagnostic/reference/torch_ops/shape_ops.py

@@ -1,10 +1,10 @@
 from typing import Optional, Tuple
 import onnx
 import torch
-from . import OpRun, OpRunTensor
+from . import OpRunKernel, OpRunTensor
 
 
-class ConstantOfShape_9(OpRun):
+class ConstantOfShape_9(OpRunKernel):
     "ConstantOfShape"
 
     @classmethod
@@ -19,8 +19,9 @@ class ConstantOfShape_9(OpRun):
         node: onnx.NodeProto,
         version: Optional[int] = None,
         device: Optional[torch.device] = None,
+        verbose: int = 0,
     ):
-        super().__init__(node, version)
+        super().__init__(node, version, verbose=verbose)
         value = self.get_attribute_tensor(node, "value")
         if value is None:
             value = torch.tensor([0], dtype=torch.float32)
@@ -37,7 +38,7 @@ class ConstantOfShape_9(OpRun):
         )
 
 
-class Expand_8(OpRun):
+class Expand_8(OpRunKernel):
     "Expand"
 
     def run(self, data: OpRunTensor, shape: OpRunTensor) -> OpRunTensor:
@@ -45,11 +46,11 @@ class Expand_8(OpRun):
         return OpRunTensor(data.tensor.expand(ishape))
 
 
-class Reshape_14(OpRun):
+class Reshape_14(OpRunKernel):
     "Reshape"
 
-    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
-        super().__init__(node, version)
+    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None, verbose: int = 0):
+        super().__init__(node, version, verbose=verbose)
         self.allowzero = self.get_attribute_int(node, "allowzero", 0)
 
     def run(self, data: OpRunTensor, shape: OpRunTensor) -> OpRunTensor:
@@ -64,9 +65,9 @@ class Reshape_14(OpRun):
         return OpRunTensor(data.tensor.reshape(ishape))
 
 
-class Shape_15(OpRun):
-    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
-        super().__init__(node, version)
+class Shape_15(OpRunKernel):
+    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None, verbose: int = 0):
+        super().__init__(node, version, verbose=verbose)
         self.start = self.get_attribute_int(node, "start", 0)
         self.end = self.get_attribute_int(node, "end", None)
 
@@ -76,9 +77,9 @@ class Shape_15(OpRun):
         return OpRunTensor(torch.tensor(sh, dtype=torch.int64), is_constant=True)
 
 
-class Split_18(OpRun):
-    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None):
-        super().__init__(node, version)
+class Split_18(OpRunKernel):
+    def __init__(self, node: onnx.NodeProto, version: Optional[int] = None, verbose: int = 0):
+        super().__init__(node, version, verbose=verbose)
         self.axis = self.get_attribute_int(node, "axis", 0)
         self.num_outputs = self.get_attribute_int(node, "num_outputs", None)
 
@@ -101,7 +102,7 @@ class Split_18(OpRun):
         return tuple(OpRunTensor(t) for t in spl)
 
 
-class Squeeze_13(OpRun):
+class Squeeze_13(OpRunKernel):
     "Squeeze"
 
     def run(self, data: OpRunTensor, axes: Optional[OpRunTensor] = None) -> OpRunTensor:
@@ -110,7 +111,7 @@ class Squeeze_13(OpRun):
         return OpRunTensor(data.tensor.squeeze(axes.as_tuple_int))
 
 
-class Unsqueeze_13(OpRun):
+class Unsqueeze_13(OpRunKernel):
     "Unsqueeze"
 
     def run(self, data: OpRunTensor, axes: OpRunTensor) -> OpRunTensor:

onnx_diagnostic/reference/torch_ops/unary_ops.py

@@ -1,85 +1,85 @@
 import torch
-from . import OpRun, OpRunTensor
+from . import OpRunKernel, OpRunTensor
 
 
-class Abs_1(OpRun):
+class Abs_1(OpRunKernel):
     """Abs"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(torch.abs(x.tensor))
 
 
-class Cos_1(OpRun):
+class Cos_1(OpRunKernel):
     """Cos"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor.cos())
 
 
-class Erf_9(OpRun):
+class Erf_9(OpRunKernel):
     """Erf"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor.erf())
 
 
-class Exp_1(OpRun):
+class Exp_1(OpRunKernel):
     """Exp"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor.exp())
 
 
-class Identity_1(OpRun):
+class Identity_1(OpRunKernel):
     "Identity"
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor)
 
 
-class Log_1(OpRun):
+class Log_1(OpRunKernel):
     """Log"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor.log())
 
 
-class Neg_1(OpRun):
+class Neg_1(OpRunKernel):
     """Neg"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(-x.tensor)
 
 
-class Not_1(OpRun):
+class Not_1(OpRunKernel):
     """Not"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(~x.tensor)
 
 
-class Reciprocal_1(OpRun):
+class Reciprocal_1(OpRunKernel):
     """REciprocal"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(1 / x.tensor)
 
 
-class Sigmoid_6(OpRun):
+class Sigmoid_6(OpRunKernel):
     """Sqrt"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(torch.sigmoid(x.tensor))
 
 
-class Sin_1(OpRun):
+class Sin_1(OpRunKernel):
     """Sin"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:
         return OpRunTensor(x.tensor.sin())
 
 
-class Sqrt_1(OpRun):
+class Sqrt_1(OpRunKernel):
     """Sqrt"""
 
     def run(self, x: OpRunTensor) -> OpRunTensor:

onnx_diagnostic/torch_export_patches/patch_module_helper.py

@@ -80,6 +80,7 @@ def known_transformers_rewritings_clamp_float16() -> Dict[str, str]:
         "AutoformerModel": "AutoformerEncoderLayer",
         "BartEncoderLayer": "BartEncoderLayer",
         "BartForConditionalGeneration": "BartEncoderLayer",
+        "BartModel": "BartEncoderLayer",
         "BigBirdPegasusForConditionalGeneration": "BigBirdPegasusEncoderLayer",
         "BigBirdPegasusForQuestionAnswering": "BigBirdPegasusEncoderLayer",
         "BigBirdPegasusForCausalLM": "BigBirdPegasusEncoderLayer",

onnx_diagnostic/torch_models/test_helper.py

@@ -278,6 +278,8 @@ def validate_model(
 ) -> Tuple[Dict[str, Union[int, float, str]], Dict[str, Any]]:
     """
     Validates a model.
+    The function can also be called through the command line
+    :ref:`l-cmd-validate`.
 
     :param model_id: model id to validate
     :param task: task used to generate the necessary inputs,
@@ -285,7 +287,8 @@ def validate_model(
         if it can be determined
     :param do_run: checks the model works with the defined inputs
     :param exporter: exporter the model using this exporter,
-        available list: ``export-strict``, ``export-nostrict``, ``onnx``
+        available list: ``export-strict``, ``export-nostrict``, ...
+        see below
     :param do_same: checks the discrepancies of the exported model
     :param verbose: verbosity level
     :param dtype: uses this dtype to check the model
@@ -322,6 +325,20 @@ def validate_model(
     information:
 
     * ``PRINT_CONFIG``: prints the model configuration
+
+    The following exporters are available:
+
+    * ``export-nostrict``: run :func:`torch.export.export` (..., strict=False)
+    * ``onnx-dynamo``: run :func:`torch.onnx.export` (..., dynamo=True),
+      models can be optimized with ``optimization`` in ``("ir", "os_ort")``
+    * ``modelbuilder``: use :epkg:`ModelBuilder` to builds the onnx model
+    * ``custom``: custom exporter (see :epkg:`experimental-experiment`),
+      models can be optimized with ``optimization`` in
+      ``("default", "default+onnxruntime", "default+os_ort", "default+onnxruntime+os_ort")``
+
+    The default runtime, :epkg:`onnxruntime` is used to validate a model and check the
+    exported model returns the same outputs as the original one, otherwise,
+    :class:`onnx_diagnostic.reference.TorchOnnxEvaluator` is used.
     """
     assert (
         not rewrite or patch
@@ -370,6 +387,12 @@ def validate_model(
         if model_options:
             print(f"[validate_model] model_options={model_options!r}")
         print(f"[validate_model] get dummy inputs with input_options={input_options}...")
+        print(
+            f"[validate_model] rewrite={rewrite}, patch={patch}, "
+            f"stop_if_static={stop_if_static}"
+        )
+        print(f"[validate_model] exporter={exporter!r}, optimization={optimization!r}")
+        print(f"[validate_model] dump_folder={dump_folder!r}")
         summary["model_id"] = model_id
         summary["model_subfolder"] = subfolder or ""
 
@@ -429,6 +452,8 @@ def validate_model(
                 print(f"[validate_model] model_rewrite={summary['model_rewrite']}")
         else:
             del data["rewrite"]
+            if verbose:
+                print("[validate_model] no rewrite")
     if os.environ.get("PRINT_CONFIG", "0") in (1, "1"):
         print("[validate_model] -- PRINT CONFIG")
         print("-- type(config)", type(data["configuration"]))
@@ -1317,13 +1342,13 @@ def call_torch_export_custom(
         "custom-nostrict",
         "custom-nostrict-default",
         "custom-nostrict-all",
-        "custom-inline",
-        "custom-strict-inline",
-        "custom-strict-default-inline",
-        "custom-strict-all-inline",
-        "custom-nostrict-inline",
-        "custom-nostrict-default-inline",
-        "custom-nostrict-all-inline",
+        "custom-noinline",
+        "custom-strict-noinline",
+        "custom-strict-default-noinline",
+        "custom-strict-all-noinline",
+        "custom-nostrict-noinline",
+        "custom-nostrict-default-noinline",
+        "custom-nostrict-all-noinline",
     }
     assert exporter in available, f"Unexpected value for exporter={exporter!r} in {available}"
     assert "model" in data, f"model is missing from data: {sorted(data)}"
@@ -1364,10 +1389,7 @@ def call_torch_export_custom(
         ),
         save_ep=(os.path.join(dump_folder, f"{exporter}.ep") if dump_folder else None),
     )
-    inline = "-inline" in exporter
-    if inline:
-        export_options.aten_as_function = set()
-
+    inline = "-noinline" not in exporter
     options = OptimizationOptions(patterns=optimization) if optimization else None
     model = data["model"]
     kws = dict(

{onnx_diagnostic-0.6.1.dist-info → onnx_diagnostic-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnx-diagnostic
-Version: 0.6.1
+Version: 0.6.3
 Summary: Investigate ONNX models
 Home-page: https://github.com/sdpython/onnx-diagnostic
 Author: Xavier Dupré

{onnx_diagnostic-0.6.1.dist-info → onnx_diagnostic-0.6.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-onnx_diagnostic/__init__.py,sha256=yCtQh931_2K3SHqvkJ_EVFSKxjtdNDc21rdCxdemDBM,173
+onnx_diagnostic/__init__.py,sha256=mRkq5dlSo05GQMct7d6mMZLb6s5T24eG_3mD5O3wBo0,173
 onnx_diagnostic/__main__.py,sha256=YmyV_Aq_ianDlHyKLHMa6h8YK3ZmFPpLVHLKjM91aCk,79
-onnx_diagnostic/_command_lines_parser.py,sha256=WSoopSHjXWEgFvzyfGe3_c-hZoStuQPQc_k08siFuf4,19211
+onnx_diagnostic/_command_lines_parser.py,sha256=pxG3dYpTTpRCjBRzFGEZm4ewb7xprZihP7fG08kWL04,19989
 onnx_diagnostic/api.py,sha256=BhCl_yCd78N7TlVtPOHjeYv1QBEy39TjZ647rcHqLh0,345
-onnx_diagnostic/doc.py,sha256=MTuT7Kxyvn7KEy84liQeFeqhugJrUQhjjpx21F72Uxw,926
+onnx_diagnostic/doc.py,sha256=O_ncetL0G4-oHIxLv8ofTIIxCT_5ESSkKxfYvgccJEc,2038
 onnx_diagnostic/ext_test_case.py,sha256=nhWz75caudvKn-svH1ppUY8uw8MoTD4cEdqMdj6PiPc,42399
 onnx_diagnostic/export/__init__.py,sha256=yEIoWiOeTwBsDhyYt2fTKuhtA0Ya1J9u9ZzMTOTWaWs,101
 onnx_diagnostic/export/dynamic_shapes.py,sha256=EHB7VoWNx8sVetvOgE1vgC7wHtIjWDLjanhbEJNpK88,39892
@@ -12,12 +12,13 @@ onnx_diagnostic/helpers/args_helper.py,sha256=SRWnqC7EENg09RZlA50B_PcdiIhdbgA4C3
 onnx_diagnostic/helpers/bench_run.py,sha256=CGA6VMJZMH2gDhVueT9ypNm4PMcjGrrGFYp08nhWj9k,16539
 onnx_diagnostic/helpers/cache_helper.py,sha256=soKjyIXa7EQgALd9PAUGIKYzXlJGoLevYiQDsxoqkQ4,8349
 onnx_diagnostic/helpers/config_helper.py,sha256=aZATKVbZuw8L56KQpwMNcqJ3Qi5OplzS_N3ETR3hmj0,3351
+onnx_diagnostic/helpers/doc_helper.py,sha256=pl5MZd3_FaE8BqQnqoBuSBxoNCFcd2OJd3eITUSku5c,5897
 onnx_diagnostic/helpers/graph_helper.py,sha256=hevQT5a7_QuriVPQcbT5qe18n99Doyl5h3-qshx1-uk,14093
 onnx_diagnostic/helpers/helper.py,sha256=oPybQruFcVLqvqLDhjphOZ8zZU1HHJWAlABMuTkAO8A,57090
 onnx_diagnostic/helpers/memory_peak.py,sha256=OT6mz0muBbBZY0pjgW2_eCk_lOtFRo-5w4jFo2Z6Kok,6380
 onnx_diagnostic/helpers/mini_onnx_builder.py,sha256=R1Vu4zHzN7GIUnbMVQzpkaXj8cCyyOweWOI9-TSgAHM,20966
-onnx_diagnostic/helpers/model_builder_helper.py,sha256=cQuC7wbl8UcU_IaSBnX_CsydKTABN6lABCWEthmpmDA,13386
-onnx_diagnostic/helpers/onnx_helper.py,sha256=chw-HB4iqGCD_16d0_BaCnreEgWYW4KeH78nh-3t2Uw,29213
+onnx_diagnostic/helpers/model_builder_helper.py,sha256=xIZmsVMFHfdtYeZHVEffBtxYObAaRPiaSmwwSKkmLwY,13502
+onnx_diagnostic/helpers/onnx_helper.py,sha256=pXXQjfyNTSUF-Kt72U4fnBDkYAnWYMxdSw8m0qk3xmE,39670
 onnx_diagnostic/helpers/ort_session.py,sha256=UgUUeUslDxEFBc6w6f3HMq_a7bn4TBlItmojqWquSj4,29281
 onnx_diagnostic/helpers/rt_helper.py,sha256=BXU_u1syk2RyM0HTFHKEiO6rHHhZW2UFPyUTVdeq8BU,4251
 onnx_diagnostic/helpers/torch_helper.py,sha256=mrmn4mBeRvMRJ9cEu7BbNG-AHq2OJfSm8dxgtzh-yQQ,31631
@@ -25,7 +26,7 @@ onnx_diagnostic/reference/__init__.py,sha256=nrd09rRuwMDBCPTSZ6kSKZXp1W9W_ExO1t9
 onnx_diagnostic/reference/evaluator.py,sha256=RzNzjFDeMe-4X51Tb22N6aagazY5ktNq-mRmPcfY5EU,8848
 onnx_diagnostic/reference/ort_evaluator.py,sha256=OaWMREF8fuJwimmONpIjQ6WxQT1X2roDsdJsgR8H_Cg,24853
 onnx_diagnostic/reference/quantized_tensor.py,sha256=5u67uS2uGacdMD5VYCbpojNjiesDlV_kO0fAJ0vUWGE,1098
-onnx_diagnostic/reference/torch_evaluator.py,sha256=uvcpDROoCDd8Rln_kieNRfraNg9j1FD39n1xiGNh3LA,20825
+onnx_diagnostic/reference/torch_evaluator.py,sha256=qAeYvSFwOCMDctc39evBEle_2bX8kuJW2QSLksofzn8,26600
 onnx_diagnostic/reference/ops/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 onnx_diagnostic/reference/ops/op_add_add_mul_mul.py,sha256=CXQVtgVrT066gDJFwxL4nDSY4G8r08XNu3EwhWqMapU,1521
 onnx_diagnostic/reference/ops/op_attention.py,sha256=ThALMDF53v3QeG1bohi0bvX2o90HZhGJbbAFOtwEHPE,2027
@@ -55,18 +56,18 @@ onnx_diagnostic/reference/ops/op_skip_layer_normalization.py,sha256=oJ7fQNx2iQh9
 onnx_diagnostic/reference/ops/op_slice.py,sha256=yRxfYBs8b7QezyyG9JHCD8MIJHij2qR2NNDpBmD3FJI,705
 onnx_diagnostic/reference/ops/op_transpose_cast.py,sha256=ifef74rvh0Yvq1Zx51B4mfnISbxV9uRg9DFjkdL1_68,361
 onnx_diagnostic/reference/ops/op_tri_matrix.py,sha256=Yn2gxAyygcwtF5Hjau9ihXDAzul0BAkdqVimVahtFBU,519
-onnx_diagnostic/reference/torch_ops/__init__.py,sha256=DzuCaiKbzJnkXprJVbVJwA7A29w6T_W1_plB_ZXDQsk,1144
-onnx_diagnostic/reference/torch_ops/_op_run.py,sha256=1nnW6Ub3jWkS-HI3--dhYNXRLqy9Ppe7F9kOeGGYRhw,10253
-onnx_diagnostic/reference/torch_ops/access_ops.py,sha256=sLiTbZUfNbATQxVpq2Rv8n6cbV4jJmQlWHUVIds_GlY,3102
-onnx_diagnostic/reference/torch_ops/binary_ops.py,sha256=774QlWFPvNtNhwxtLb-d28jhTO8U0cthoJKHjM89WrA,2641
-onnx_diagnostic/reference/torch_ops/controlflow_ops.py,sha256=EsiMFmLHxFHprCSz9OLlYxo3sMaF9czj-4NOFkHZH5c,4362
-onnx_diagnostic/reference/torch_ops/generator_ops.py,sha256=oxoeg7YxQiEaC20qOFI-BwQuBtkxN7y9XD9o3gPjPkI,871
-onnx_diagnostic/reference/torch_ops/nn_ops.py,sha256=wzXuBfMsA1aQPQDi4qAv0oZ8kB1qwt4reQ8LiWg1ulo,6906
-onnx_diagnostic/reference/torch_ops/other_ops.py,sha256=bIZsyqrRfTI1xiC4WvV5LcQawsy4gqqci9EgOO9rk9c,3766
-onnx_diagnostic/reference/torch_ops/reduce_ops.py,sha256=oeA-NNJ1MFQD7xG5RD61_3SQhXvpJpNXKZ-_0rFikjE,5020
-onnx_diagnostic/reference/torch_ops/sequence_ops.py,sha256=q1bwFI1zUO69KrOOAbNFqBhz1s9YkqxV1kCaBmgIoYI,2215
-onnx_diagnostic/reference/torch_ops/shape_ops.py,sha256=1pR-3EwhuMpAE-B2TUJs83GxDGITaLBHKxExJ-Ie1EE,4095
-onnx_diagnostic/reference/torch_ops/unary_ops.py,sha256=T3EtQW0p_kNN-OhjVsk1STZ0uWxXyB-rYO-lIp3gXuU,1630
+onnx_diagnostic/reference/torch_ops/__init__.py,sha256=eZ7FBYH9Ta3BC0f7BJQAnIqiebqRkCt2_T3ktpVV6iQ,1150
+onnx_diagnostic/reference/torch_ops/_op_run.py,sha256=EEUIwfbRldEFDhULquYhk9x5ZDa9t6f2mKJ1sM__D6A,10517
+onnx_diagnostic/reference/torch_ops/access_ops.py,sha256=Zfs5OF03PV1CqlCqKI5VV-c4MY3KyQxmO7QZksxQjX8,3274
+onnx_diagnostic/reference/torch_ops/binary_ops.py,sha256=-KxMcCYGDTcZyOss9qU1nU0rmdyg9SdVHJQohseSTcQ,2653
+onnx_diagnostic/reference/torch_ops/controlflow_ops.py,sha256=uOEmzbM4nR2FwZQ8UikwEaHih3yw6T24D_VLYkr5RSU,4518
+onnx_diagnostic/reference/torch_ops/generator_ops.py,sha256=dqqFvhkazVxRUDYhO2t-Z1etqoUt30LeX-8Pb9ZcbaM,926
+onnx_diagnostic/reference/torch_ops/nn_ops.py,sha256=TeFxQEiTezx9UUzu82ToCLUnUOU56kKv6X81RdZ8UC8,7238
+onnx_diagnostic/reference/torch_ops/other_ops.py,sha256=FnCY60mhdrzrsiHgvN-XpFRHYUpI0gIRqxgVK5J_na0,3995
+onnx_diagnostic/reference/torch_ops/reduce_ops.py,sha256=9gFfraPTQbe_ZEUNCUis1JSmA5dj4tSzjAOpZPJKG4Y,5102
+onnx_diagnostic/reference/torch_ops/sequence_ops.py,sha256=3EiVKpGfN4d1Iry4hgnr3MIJyEEKUrAIDgmRGsUXXa0,2297
+onnx_diagnostic/reference/torch_ops/shape_ops.py,sha256=pJrNR2UB4PlWl6cv4EDl1uGl8YTBUUMQkhJcsh5K4sA,4291
+onnx_diagnostic/reference/torch_ops/unary_ops.py,sha256=E8Ys1eZsOTsucBKoXb1_Kl5LbBDygniDvW2BvN4IPMo,1708
 onnx_diagnostic/tasks/__init__.py,sha256=5XXM-rv-Hk2gSHvqsww9DzVd9mcRifacgcPgvPCjnDM,2412
 onnx_diagnostic/tasks/automatic_speech_recognition.py,sha256=oRoYy56M0Yv_WOcn1hJXv-R9wgHkJ8rbym7j7y8oslw,6851
 onnx_diagnostic/tasks/feature_extraction.py,sha256=V-T5NpZ6EimOz00weWWxGfksZ9jQ5ZQyaP-mxuCEuJo,2223
@@ -87,7 +88,7 @@ onnx_diagnostic/torch_export_patches/onnx_export_serialization.py,sha256=l5HvE_F
 onnx_diagnostic/torch_export_patches/patch_expressions.py,sha256=vr4tt61cbDnaaaduzMj4UBZ8OUtr6GfDpIWwOYqjWzs,3213
 onnx_diagnostic/torch_export_patches/patch_inputs.py,sha256=9b4pmyT00BwLqi7WG-gliep1RUy3gXEgW6BDnlSSA-M,7689
 onnx_diagnostic/torch_export_patches/patch_module.py,sha256=R2d9IHM-RwsBKDsxuBIJnEqMoxbS9gd4YWFGG2wwV5A,39881
-onnx_diagnostic/torch_export_patches/patch_module_helper.py,sha256=sIvu4E9BsCB8f-KlM4xykR19mflDfLGaiel6nb9ZGx8,6926
+onnx_diagnostic/torch_export_patches/patch_module_helper.py,sha256=2U0AdyZuU0W54QTdE7tY7imVzMnpQ5091ADNtTCkT8Y,6967
 onnx_diagnostic/torch_export_patches/eval/__init__.py,sha256=VtkQB1o3Q2Fh99OOF6vQ2dynkhwzx2Wx6oB-rRbvTI0,23954
 onnx_diagnostic/torch_export_patches/eval/model_cases.py,sha256=DTvdHPtNQh25Akv5o3D4Jxf1L1-SJ7w14tgvj8AAns8,26577
 onnx_diagnostic/torch_export_patches/patches/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -95,7 +96,7 @@ onnx_diagnostic/torch_export_patches/patches/patch_torch.py,sha256=KaZ8TjDa9ATgT
 onnx_diagnostic/torch_export_patches/patches/patch_transformers.py,sha256=Hf-U50vzgzJ4iUjS2LAYkbfmzCEwX80Dzvdrr-Rhlp0,26456
 onnx_diagnostic/torch_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 onnx_diagnostic/torch_models/llms.py,sha256=soyg4yC87ptGoeulJhKqw5opGmuLvH1pn_ZDXZ4Jr8E,90
-onnx_diagnostic/torch_models/test_helper.py,sha256=72HE_fY0m9TNly1w_b2dD-P67eSLJreOMYtT6W6CLfQ,58704
+onnx_diagnostic/torch_models/test_helper.py,sha256=tt6bgLjGRayzvkXrTelKHTjr7XU9BvhX7uE4XJq6H6o,59927
 onnx_diagnostic/torch_models/hghub/__init__.py,sha256=vi1Q7YHdddj1soiBN42MSvJdFqe2_KUoWafHISjwOu8,58
 onnx_diagnostic/torch_models/hghub/hub_api.py,sha256=BgM_p57Q0gT9GOhdrmOYcnbuTTzCWp80jS4OQqWwFhs,9990
 onnx_diagnostic/torch_models/hghub/hub_data.py,sha256=885wKyZkdM-Qp5Sg6C9Ol1dxigmA8FYAko-Ys08sppo,8096
@@ -107,8 +108,8 @@ onnx_diagnostic/torch_models/untrained/llm_tiny_llm.py,sha256=7N3fGvT_4Mn4NbIo0Q
 onnx_diagnostic/torch_onnx/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 onnx_diagnostic/torch_onnx/runtime_info.py,sha256=1g9F_Jf9AAgYQU4stbsrFXwQl-30mWlQrFbQ7val8Ps,9268
 onnx_diagnostic/torch_onnx/sbs.py,sha256=1EL25DeYFzlBSiFG_XjePBLvsiItRXbdDrr5-QZW2mA,16878
-onnx_diagnostic-0.6.1.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
-onnx_diagnostic-0.6.1.dist-info/METADATA,sha256=j0i2pgpY_6bD7IFlqqdmIZSVSvv67Q-7X4N4fih-2JQ,6643
-onnx_diagnostic-0.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-onnx_diagnostic-0.6.1.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
-onnx_diagnostic-0.6.1.dist-info/RECORD,,
+onnx_diagnostic-0.6.3.dist-info/licenses/LICENSE.txt,sha256=Vv6TXglX6Rc0d-f8aREhayhT-6PMQXEyOmI2NKlUCMc,1045
+onnx_diagnostic-0.6.3.dist-info/METADATA,sha256=eJxj0KTPv1rXf-3T9KImWIF-u8g7wHXBGZm5zvXM7V8,6643
+onnx_diagnostic-0.6.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+onnx_diagnostic-0.6.3.dist-info/top_level.txt,sha256=KwNkXewmcobM3ZT1DJLVWH6ebJzA5qKg7cWqKfpGNT4,16
+onnx_diagnostic-0.6.3.dist-info/RECORD,,