onnxslim 0.1.79__py3-none-any.whl → 0.1.81__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 (11) hide show
  1. onnxslim/core/optimization/dead_node_elimination.py +4 -0
  2. onnxslim/third_party/symbolic_shape_infer.py +5 -4
  3. onnxslim/utils.py +33 -26
  4. onnxslim/version.py +3 -1
  5. {onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/METADATA +18 -28
  6. {onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/RECORD +9 -11
  7. {onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/WHEEL +1 -2
  8. onnxslim-0.1.79.dist-info/top_level.txt +0 -1
  9. onnxslim-0.1.79.dist-info/zip-safe +0 -1
  10. {onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/entry_points.txt +0 -0
  11. {onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/licenses/LICENSE +0 -0
onnxslim/core/optimization/dead_node_elimination.py CHANGED
@@ -53,6 +53,10 @@ def dead_node_elimination(graph, is_subgraph=False):
53
53
  node.inputs.pop(1)
54
54
  node.inputs.insert(1, reshape_const)
55
55
  logger.debug(f"replacing {node.op} op: {node.name}")
56
+ elif node.op == "Slice":
57
+ if node.inputs[0].shape and node.outputs[0].shape and node.inputs[0].shape == node.outputs[0].shape:
58
+ node.erase()
59
+ logger.debug(f"removing {node.op} op: {node.name}")
56
60
  elif node.op == "Mul":
57
61
  if (isinstance(node.inputs[1], Constant) and isinstance(node.inputs[0], Variable)) or (
58
62
  isinstance(node.inputs[0], Constant) and isinstance(node.inputs[1], Variable)
onnxslim/third_party/symbolic_shape_infer.py CHANGED
@@ -1865,8 +1865,10 @@ class SymbolicShapeInference:
1865
1865
  else:
1866
1866
  scales = list(scales)
1867
1867
  new_sympy_shape = [
1868
- (round(d * (end - start) * scale))
1869
- for d, start, end, scale in zip(input_sympy_shape, roi_start, roi_end, scales)
1868
+ sympy.floor(d * (end - start) * scale + sympy.Rational(1, 2))
1869
+ for d, start, end, scale in zip(
1870
+ input_sympy_shape, roi_start, roi_end, scales
1871
+ )
1870
1872
  ]
1871
1873
  self._update_computed_dims(new_sympy_shape)
1872
1874
  else:
@@ -2117,9 +2119,8 @@ class SymbolicShapeInference:
2117
2119
  if len(e.free_symbols) == 1:
2118
2120
  if try_solve((e - new_sympy_shape[i]) >= 0, next(iter(e.free_symbols))) is None:
2119
2121
  logger.warning(
2120
- f"Unable to determine if {e} <= {new_sympy_shape[i]}, treat as equal"
2122
+ f"Unable to solve if {e} <= {new_sympy_shape[i]}, treat as not equal"
2121
2123
  )
2122
- e = new_sympy_shape[i]
2123
2124
  else:
2124
2125
  logger.warning(f"Unable to determine if {e} <= {new_sympy_shape[i]}, treat as equal")
2125
2126
  e = new_sympy_shape[i]
onnxslim/utils.py CHANGED
@@ -664,32 +664,39 @@ def is_onnxruntime_available():
664
664
  def check_onnx_compatibility():
665
665
  """Ensure ONNX Runtime and ONNX versions are compatible for model inference."""
666
666
  compatibility_dict = {
667
- "1.20": "1.16",
668
- "1.19": "1.16",
669
- "1.18": "1.16",
670
- "1.17": "1.15",
671
- "1.16": "1.14.1",
672
- "1.15": "1.14",
673
- "1.14": "1.13",
674
- "1.13": "1.12",
675
- "1.12": "1.12",
676
- "1.11": "1.11",
677
- "1.10": "1.10",
678
- "1.9": "1.10",
679
- "1.8": "1.9",
680
- "1.7": "1.8",
681
- "1.6": "1.8",
682
- "1.5": "1.7",
683
- "1.4": "1.7",
684
- "1.3": "1.7",
685
- "1.2": "1.6",
686
- "1.1": "1.6",
687
- "1.0": "1.6",
688
- "0.5": "1.5",
689
- "0.4": "1.5",
690
- "0.3": "1.4",
691
- "0.2": "1.3",
692
- "0.1": "1.3",
667
+ # ONNX Runtime version -> ONNX version (max supported)
668
+ # 2024-2025 releases
669
+ "1.23": "1.18", # opset 23, released 2025-09
670
+ "1.22": "1.17", # opset 22, released 2025-05
671
+ "1.21": "1.17", # opset 22, released 2025-03
672
+ "1.20": "1.16", # opset 21, released 2024-11
673
+ "1.19": "1.16", # opset 21, released 2024-08
674
+ "1.18": "1.16", # opset 21, released 2024-05
675
+ "1.17": "1.15", # opset 20, released 2024-01
676
+ # 2023 releases
677
+ "1.16": "1.14.1", # opset 19, released 2023-09
678
+ "1.15": "1.14", # opset 19, released 2023-05
679
+ "1.14": "1.13", # opset 18, released 2023-03
680
+ "1.13": "1.12", # opset 17, released 2022-11
681
+ # 2022 and earlier
682
+ "1.12": "1.12", # opset 17
683
+ "1.11": "1.11", # opset 16
684
+ "1.10": "1.10", # opset 15
685
+ "1.9": "1.10", # opset 15
686
+ "1.8": "1.9", # opset 14
687
+ "1.7": "1.8", # opset 13
688
+ "1.6": "1.8", # opset 13
689
+ "1.5": "1.7", # opset 12
690
+ "1.4": "1.7", # opset 12
691
+ "1.3": "1.7", # opset 12
692
+ "1.2": "1.6", # opset 11
693
+ "1.1": "1.6", # opset 11
694
+ "1.0": "1.6", # opset 11
695
+ "0.5": "1.5", # opset 10
696
+ "0.4": "1.5", # opset 10
697
+ "0.3": "1.4", # opset 9
698
+ "0.2": "1.3", # opset 8
699
+ "0.1": "1.3", # opset 8
693
700
  }
694
701
  import onnx
695
702
  import onnxruntime
onnxslim/version.py CHANGED
@@ -1 +1,3 @@
1
- __version__ = "0.1.79"
1
+ from importlib.metadata import version
2
+
3
+ __version__ = version("onnxslim")
{onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/METADATA RENAMED
@@ -1,38 +1,28 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: onnxslim
3
- Version: 0.1.79
3
+ Version: 0.1.81
4
4
  Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
5
- Home-page: https://github.com/inisis/OnnxSlim
6
- Author: inisis
7
- Author-email: desmond.yao@buaa.edu.cn
8
- License: MIT
9
- Project-URL: Bug Tracker, https://github.com/inisis/OnnxSlim/issues
10
- Classifier: Programming Language :: Python :: 3
11
- Classifier: License :: OSI Approved :: MIT License
5
+ Project-URL: homepage, https://github.com/inisis/OnnxSlim
6
+ Project-URL: issues, https://github.com/inisis/OnnxSlim/issues
7
+ Author-email: inisis <desmond.yao@buaa.edu.cn>
8
+ License-Expression: MIT
9
+ License-File: LICENSE
12
10
  Classifier: Intended Audience :: Developers
11
+ Classifier: License :: OSI Approved :: MIT License
12
+ Classifier: Programming Language :: Python :: 3
13
13
  Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
14
- Requires-Python: >=3.6
15
- Description-Content-Type: text/markdown
16
- License-File: LICENSE
14
+ Requires-Python: >=3.8
15
+ Requires-Dist: colorama
16
+ Requires-Dist: ml-dtypes
17
17
  Requires-Dist: onnx
18
- Requires-Dist: sympy>=1.13.1
19
18
  Requires-Dist: packaging
20
- Requires-Dist: colorama
21
- Requires-Dist: ml_dtypes
22
- Dynamic: author
23
- Dynamic: author-email
24
- Dynamic: classifier
25
- Dynamic: description
26
- Dynamic: description-content-type
27
- Dynamic: home-page
28
- Dynamic: license
29
- Dynamic: license-file
30
- Dynamic: project-url
31
- Dynamic: requires-dist
32
- Dynamic: requires-python
33
- Dynamic: summary
34
-
35
- # OnnxSlim
19
+ Requires-Dist: sympy>=1.13.1
20
+ Description-Content-Type: text/markdown
21
+
22
+ <p align="center">
23
+ <img src="assets/logo/onnxslim-banner-light.svg#gh-light-mode-only" alt="OnnxSlim Logo" width="640"/>
24
+ <img src="assets/logo/onnxslim-banner.svg#gh-dark-mode-only" alt="OnnxSlim Logo" width="640"/>
25
+ </p>
36
26
 
37
27
  <p align="center">
38
28
  <a href="https://pypi.org/project/onnxslim">
{onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/RECORD RENAMED
@@ -1,13 +1,13 @@
1
1
  onnxslim/__init__.py,sha256=ECHGdxzg4b-4SZaPhxM_KulBi-xDbVcVUbpJc8i6a60,571
2
2
  onnxslim/__main__.py,sha256=FgDcl6xX8kV_52rB-jPVsmGqidlVhkpe_YhXK75-nFU,75
3
3
  onnxslim/argparser.py,sha256=pFv3nEZH2BiHO9ejS4Iq5ZuZ3GrpdyRQJypAyR0xF7w,8942
4
- onnxslim/utils.py,sha256=Z39vRKAtD7o18NFbf3Qrws9xDg81uBJ9F-RYEFAfqM8,28095
5
- onnxslim/version.py,sha256=wrT29P9p_epbXQwXDsdC8_jeLRCegcuNp7-xofsm8uI,23
4
+ onnxslim/utils.py,sha256=-ZWE2TnovvNMW2UlASgWtD44u_VbMCjRSXNGx8HgTAQ,28909
5
+ onnxslim/version.py,sha256=6_EhMra7UU7XSDG6tfWQrxITEFhtUFdFuQyCRcBAScQ,74
6
6
  onnxslim/cli/__init__.py,sha256=kxK27cDgWotBOWRs86rbRQf_dtmniKr1GZJeasxfESE,42
7
7
  onnxslim/cli/_main.py,sha256=jEKv9q7y_y9g0GsrfXcnk_wyMVej6jhe9QNPChE7yTs,5550
8
8
  onnxslim/core/__init__.py,sha256=uDg-Eu29Ezb3txwZf5mN0zQRVuqF-K9BvktE8WBYS4E,8825
9
9
  onnxslim/core/optimization/__init__.py,sha256=6w7VEhsfd7MhCSLmpAMLfxaEq9pf4YYHhNndggNLZCg,5034
10
- onnxslim/core/optimization/dead_node_elimination.py,sha256=JoQsWb304NSqjssFTvlbOog1N-2KLitZ-alrfex3Q88,7558
10
+ onnxslim/core/optimization/dead_node_elimination.py,sha256=B5hpxwLrdWD5uFt_r87uOgSX-Ul81-sNJlYowa4jU_Y,7801
11
11
  onnxslim/core/optimization/subexpression_elimination.py,sha256=wauPADo0msVifYGgJ_fiNmPSe7LrJ7A-D3ur5z1wif8,2838
12
12
  onnxslim/core/optimization/weight_tying.py,sha256=uAz_2AJa-s-budTcDXIr6ZcME1hXPdm54CzR-u850X4,2347
13
13
  onnxslim/core/pattern/__init__.py,sha256=az4HayGp9WybHMiijr-VfbhzPIfRc-51JK8J1_Q-5Yo,9953
@@ -30,7 +30,7 @@ onnxslim/core/pattern/fusion/reduce.py,sha256=dMC7CPlFglrJxugsJWjcc-jQCIa_GIbW1y
30
30
  onnxslim/misc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
31
31
  onnxslim/misc/tabulate.py,sha256=Pg5uU0UP18HbwG-c8LlA82LbIb_5JWQeuIB1AnturbM,99695
32
32
  onnxslim/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
33
- onnxslim/third_party/symbolic_shape_infer.py,sha256=lMqHUSC1c6J3j_cLC5264RxvJrhV8rrephsxE4d-4Zc,152724
33
+ onnxslim/third_party/symbolic_shape_infer.py,sha256=lBNcfgNUVG24VrgP_VzIi0K_5RAhF3vatuETv09Wv5Y,152738
34
34
  onnxslim/third_party/_sympy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
35
35
  onnxslim/third_party/_sympy/functions.py,sha256=s3pKzyjYCKnvlddLFR_H8UmbbcdMB51PRxqhe9zGI9E,8876
36
36
  onnxslim/third_party/_sympy/numbers.py,sha256=WOCwRg5JiIfUQCXKAUUuWis8rxdc7LLthUf3Ax1jG8I,11535
@@ -56,10 +56,8 @@ onnxslim/third_party/onnx_graphsurgeon/logger/logger.py,sha256=L12rrwn33RHH-2WLv
56
56
  onnxslim/third_party/onnx_graphsurgeon/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
57
57
  onnxslim/third_party/onnx_graphsurgeon/util/exception.py,sha256=KrsHbKEQ4237UbjlODsUzvkXoAY72LZi23ApBeFANWg,786
58
58
  onnxslim/third_party/onnx_graphsurgeon/util/misc.py,sha256=kyxInD2SCRLU4wHMeiDEYEHB3871fGks6kQTuF9uATY,8960
59
- onnxslim-0.1.79.dist-info/licenses/LICENSE,sha256=oHZXw-yrBwdNVGu4JtlZhMgmQHKIZ7BJJlJdhu1HKvI,1062
60
- onnxslim-0.1.79.dist-info/METADATA,sha256=LUuyQEROqRtF488QJMNJCL5bJF9-AalG1fe1egGMb9A,10247
61
- onnxslim-0.1.79.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
62
- onnxslim-0.1.79.dist-info/entry_points.txt,sha256=O2QgceCVeGeRhnxRSDRcGiFd0ZNfElwrTiRo1W2V7KA,47
63
- onnxslim-0.1.79.dist-info/top_level.txt,sha256=EWFTb99i0kc6cC9akqNKp88ipzg17_VZzYN7z1kQNlA,9
64
- onnxslim-0.1.79.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
65
- onnxslim-0.1.79.dist-info/RECORD,,
59
+ onnxslim-0.1.81.dist-info/METADATA,sha256=zE4GD0vim78Jk_LYDhUxmjX4t3U9m4eewdZUCyQR76c,10216
60
+ onnxslim-0.1.81.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
61
+ onnxslim-0.1.81.dist-info/entry_points.txt,sha256=O2QgceCVeGeRhnxRSDRcGiFd0ZNfElwrTiRo1W2V7KA,47
62
+ onnxslim-0.1.81.dist-info/licenses/LICENSE,sha256=oHZXw-yrBwdNVGu4JtlZhMgmQHKIZ7BJJlJdhu1HKvI,1062
63
+ onnxslim-0.1.81.dist-info/RECORD,,
{onnxslim-0.1.79.dist-info → onnxslim-0.1.81.dist-info}/WHEEL RENAMED
@@ -1,5 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: hatchling 1.28.0
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
-
onnxslim-0.1.79.dist-info/top_level.txt DELETED
@@ -1 +0,0 @@
1
- onnxslim
onnxslim-0.1.79.dist-info/zip-safe DELETED
@@ -1 +0,0 @@
1
-