onnxslim 0.1.81__tar.gz → 0.1.82__tar.gz

{onnxslim-0.1.81 → onnxslim-0.1.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxslim
-Version: 0.1.81
+Version: 0.1.82
 Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
 Project-URL: homepage, https://github.com/inisis/OnnxSlim
 Project-URL: issues, https://github.com/inisis/OnnxSlim/issues

{onnxslim-0.1.81 → onnxslim-0.1.82}/onnxslim/core/optimization/dead_node_elimination.py

@@ -53,10 +53,10 @@ def dead_node_elimination(graph, is_subgraph=False):
                     node.inputs.pop(1)
                     node.inputs.insert(1, reshape_const)
                     logger.debug(f"replacing {node.op} op: {node.name}")
-        elif node.op == "Slice":
-            if node.inputs[0].shape and node.outputs[0].shape and node.inputs[0].shape == node.outputs[0].shape:
-                node.erase()
-                logger.debug(f"removing {node.op} op: {node.name}")
+        # elif node.op == "Slice":
+        #     if node.inputs[0].shape and node.outputs[0].shape and node.inputs[0].shape == node.outputs[0].shape and all(isinstance(item, int) for item in node.inputs[0].shape):
+        #         node.erase()
+        #         logger.debug(f"removing {node.op} op: {node.name}")
         elif node.op == "Mul":
             if (isinstance(node.inputs[1], Constant) and isinstance(node.inputs[0], Variable)) or (
                 isinstance(node.inputs[0], Constant) and isinstance(node.inputs[1], Variable)

{onnxslim-0.1.81 → onnxslim-0.1.82}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "onnxslim"
-version = "0.1.81"
+version = "0.1.82"
 description = "OnnxSlim: A Toolkit to Help Optimize Onnx Model"
 authors = [{ name = "inisis", email = "desmond.yao@buaa.edu.cn" }]
 license = "MIT"

{onnxslim-0.1.81 → onnxslim-0.1.82}/tests/test_dead_node_elimination.py

@@ -366,34 +366,34 @@ class TestDeadNodeElimination:
         # Split node with single output should be eliminated
         assert final_node_count <= initial_node_count
 
-    def test_noop_slice_elimination(self, request):
-        """Test that Slice nodes that don't change the tensor are eliminated.
+    # def test_noop_slice_elimination(self, request):
+    #     """Test that Slice nodes that don't change the tensor are eliminated.
 
-        Construct the ONNX graph manually to ensure a `Slice` node exists
-        with starts/ends/axes/constants that represent a no-op slice.
-        """
+    #     Construct the ONNX graph manually to ensure a `Slice` node exists
+    #     with starts/ends/axes/constants that represent a no-op slice.
+    #     """
 
-        # Create a simple graph with an explicit Slice node
-        input_tensor = gs.Variable(name="input", dtype=np.float32, shape=(1, 512, 8, 8))
-        output_tensor = gs.Variable(name="output", dtype=np.float32, shape=(1, 512, 8, 8))
+    #     # Create a simple graph with an explicit Slice node
+    #     input_tensor = gs.Variable(name="input", dtype=np.float32, shape=(1, 512, 8, 8))
+    #     output_tensor = gs.Variable(name="output", dtype=np.float32, shape=(1, 512, 8, 8))
 
-        starts = gs.Constant(name="starts", values=np.array([0, 0], dtype=np.int64))
-        ends = gs.Constant(name="ends", values=np.array([8, 8], dtype=np.int64))
-        axes = gs.Constant(name="axes", values=np.array([2, 3], dtype=np.int64))
+    #     starts = gs.Constant(name="starts", values=np.array([0, 0], dtype=np.int64))
+    #     ends = gs.Constant(name="ends", values=np.array([8, 8], dtype=np.int64))
+    #     axes = gs.Constant(name="axes", values=np.array([2, 3], dtype=np.int64))
 
-        slice_node = gs.Node(op="Slice", inputs=[input_tensor, starts, ends, axes], outputs=[output_tensor])
+    #     slice_node = gs.Node(op="Slice", inputs=[input_tensor, starts, ends, axes], outputs=[output_tensor])
 
-        graph = gs.Graph(
-            inputs=[input_tensor],
-            outputs=[output_tensor],
-            nodes=[slice_node],
-        )
+    #     graph = gs.Graph(
+    #         inputs=[input_tensor],
+    #         outputs=[output_tensor],
+    #         nodes=[slice_node],
+    #     )
 
-        dead_node_elimination(graph)
-        graph.cleanup().toposort()
+    #     dead_node_elimination(graph)
+    #     graph.cleanup().toposort()
 
-        # Slice that does not change shape should be eliminated
-        assert not any(node.op == "Slice" for node in graph.nodes)
+    #     # Slice that does not change shape should be eliminated
+    #     assert not any(node.op == "Slice" for node in graph.nodes)
 
     def test_check_shape_function(self):
         """Test the check_shape helper function."""

{onnxslim-0.1.81 → onnxslim-0.1.82}/tests/test_modelzoo.py

@@ -193,6 +193,21 @@ class TestModelZoo:
             assert len(summary.op_info["/avgpool/GlobalAveragePool"].outputs) == 1
             assert summary.op_info["/avgpool/GlobalAveragePool"].outputs[0].shape == (1, 2048, 1, 1)
 
+    def test_yolo11n(self, request):
+        name = request.node.originalname[len("test_") :]
+        filename = f"{MODELZOO_PATH}/{name}/{name}.onnx"
+
+        with tempfile.TemporaryDirectory() as tempdir:
+            slim(filename, os.path.join(tempdir, f"{name}_slim.onnx"))
+            input = np.zeros((1, 3, 256, 256), dtype=np.float32)
+
+            ort_sess = ort.InferenceSession(os.path.join(tempdir, f"{name}_slim.onnx"))
+            ort_sess.run(None, {"images": input})
+
+            summary = summarize_model(os.path.join(tempdir, f"{name}_slim.onnx"), tag=request.node.name)
+            assert summary.op_type_counts["Add"] == 25
+            assert summary.op_type_counts["Concat"] == 44
+
 
 if __name__ == "__main__":
     import sys