onnx-diagnostic 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (14) hide show
  1. onnx_diagnostic/__init__.py +1 -1
  2. onnx_diagnostic/export/dynamic_shapes.py +1 -1
  3. onnx_diagnostic/ext_test_case.py +530 -546
  4. onnx_diagnostic/helpers.py +10 -4
  5. onnx_diagnostic/ort_session.py +1 -0
  6. onnx_diagnostic/torch_export_patches/onnx_export_errors.py +67 -117
  7. onnx_diagnostic/torch_export_patches/onnx_export_serialization.py +0 -18
  8. onnx_diagnostic/torch_export_patches/patches/patch_transformers.py +342 -91
  9. onnx_diagnostic/torch_test_helper.py +16 -3
  10. {onnx_diagnostic-0.2.0.dist-info → onnx_diagnostic-0.2.1.dist-info}/METADATA +4 -2
  11. {onnx_diagnostic-0.2.0.dist-info → onnx_diagnostic-0.2.1.dist-info}/RECORD +14 -14
  12. {onnx_diagnostic-0.2.0.dist-info → onnx_diagnostic-0.2.1.dist-info}/WHEEL +1 -1
  13. {onnx_diagnostic-0.2.0.dist-info → onnx_diagnostic-0.2.1.dist-info}/licenses/LICENSE.txt +0 -0
  14. {onnx_diagnostic-0.2.0.dist-info → onnx_diagnostic-0.2.1.dist-info}/top_level.txt +0 -0
onnx_diagnostic/__init__.py CHANGED
@@ -3,5 +3,5 @@ Investigates onnx models.
3
3
  Functions, classes to dig into a model when this one is right, slow, wrong...
4
4
  """
5
5
 
6
- __version__ = "0.2.0"
6
+ __version__ = "0.2.1"
7
7
  __author__ = "Xavier Dupré"
onnx_diagnostic/export/dynamic_shapes.py CHANGED
@@ -311,7 +311,7 @@ class ModelInputs:
311
311
  shapes[i] = self.guess_dynamic_shape_object(*[o[i] for o in objs])
312
312
  return shapes
313
313
 
314
- if obj.__class__.__name__ in ("DynamicCache", "patched_DynamicCache"):
314
+ if obj.__class__.__name__ == "DynamicCache":
315
315
  kc = set(len(o.key_cache) for o in objs)
316
316
  assert (
317
317
  len(kc) == 1