onnxscript 0.7.2.dev20260710__tar.gz → 0.7.2.dev20260714__tar.gz

{onnxscript-0.7.2.dev20260710/onnxscript.egg-info → onnxscript-0.7.2.dev20260714}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.2.dev20260710
+Version: 0.7.2.dev20260714
 Summary: Naturally author ONNX functions and models using a subset of Python
 Author-email: Microsoft Corporation <onnx@microsoft.com>
 License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
         
 Project-URL: Homepage, https://microsoft.github.io/onnxscript/
 Project-URL: Repository, https://github.com/microsoft/onnxscript
-Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/e6fad72c99db6a5639ab8ee9bd9c04c9039c1cb0
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/d3489d49a9d63d039545e74c0ce80cbbbe241236
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers

{onnxscript-0.7.2.dev20260710 → onnxscript-0.7.2.dev20260714}/onnxscript/function_libs/torch_lib/ops/core.py

@@ -8290,12 +8290,15 @@ def aten_repeat_interleave_self_int(
     .. code-block:: python
 
         x = torch.tensor([[0, 1, 2], [3, 4, 5]])
-        x.repeat((1, 2)).reshape((-1, t.shape[1]))
+        x.repeat((1, 2)).reshape((-1, x.shape[1]))
     """
     if dim is None:
-        raise NotImplementedError("No conversion available yet when dim is None.")
+        self = op.Reshape(self, [-1])
+        dim = 0
+        self_rank = 1
+    else:
+        self_rank = len(self.shape)
 
-    self_rank = len(self.shape)
     pos_dim = (dim + self_rank) % self_rank
     unsqueezed = op.Unsqueeze(self, [pos_dim + 1])
     if isinstance(repeats, int):
@@ -8311,8 +8314,6 @@ def aten_repeat_interleave_self_int(
             axis=0,
         )
     tiled = op.Expand(unsqueezed, tile_repeat)
-    if self_rank == 1:
-        return op.Identity(tiled)
     final_shape = op.Concat(
         op.Shape(self, start=0, end=dim),
         op.Constant(value_ints=[-1]),

{onnxscript-0.7.2.dev20260710 → onnxscript-0.7.2.dev20260714}/onnxscript/optimizer/_constant_folding.py

@@ -564,6 +564,21 @@ def size(node: ir.Node, op, state: OptimizerState) -> ReturnValue:
     return op.Constant(value_int=size)
 
 
+def _move_initializers_to_graph(src: ir.Graph, dst: ir.Graph) -> None:
+    """Move all initializers from src graph to dst graph, ensuring name uniqueness."""
+    counter: dict[str, int] = {}
+    for name in list(src.initializers):
+        initializer = src.initializers.pop(name)
+        # Ensure name uniqueness in the destination graph
+        new_name = name
+        while new_name in dst.initializers:
+            counter[name] = counter.get(name, 0) + 1
+            new_name = f"{name}_{counter[name]}"
+        if new_name != name:
+            initializer.name = new_name
+        dst.register_initializer(initializer)
+
+
 @register("If")
 def if_op(node: ir.Node, op, state: OptimizerState) -> ReturnValue:
     cond_input = _get_input(node, 0)
@@ -601,7 +616,11 @@ def if_op(node: ir.Node, op, state: OptimizerState) -> ReturnValue:
             # Avoid name collision.
             sub_node.name = f"{node.name}_{sub_node.name}"
 
-        # TODO: we should handle initializers as well!
+        # Move initializers from the subgraph to the main graph to avoid losing them.
+        main_graph = node.graph
+        if main_graph is not None:
+            _move_initializers_to_graph(graph, main_graph)
+
         return Replacement(formal_outs, graph_nodes)
     return None
 
@@ -863,11 +882,34 @@ def split_to_sequence(node: ir.Node, op, state: OptimizerState) -> ReturnValue:
         split_dimension_size = shape[axis]
         if not isinstance(split_dimension_size, int):
             return None
-        num_outputs = math.ceil(split_dimension_size / split_value.item())
+        split_size = int(split_value.item())
+        if split_size <= 0:
+            # Invalid split size; bail out instead of raising.
+            return None
+        num_outputs = math.ceil(split_dimension_size / split_size)
         split_outputs = [f"{output.name}_split_{i}" for i in range(num_outputs)]
-        split_values = op.Split(
-            input, axis=axis, num_outputs=num_outputs, _outputs=split_outputs
-        )
+        if split_dimension_size % split_size != 0:
+            # Uneven split: the last chunk is smaller. We must pass explicit split
+            # sizes to Split, because Split with only num_outputs would do an
+            # equal (or near-equal) split ignoring the original chunk size.
+            # NOTE: Split accepts either the `split` input or the `num_outputs`
+            # attribute, but not both, so num_outputs is omitted here.
+            remainder = split_dimension_size - (num_outputs - 1) * split_size
+            explicit_split_sizes = [split_size] * (num_outputs - 1) + [remainder]
+            explicit_split = op.Constant(
+                value_ints=explicit_split_sizes,
+                _outputs=[f"{output.name}_split_sizes"],
+            )
+            split_values = op.Split(
+                input,
+                explicit_split,
+                axis=axis,
+                _outputs=split_outputs,
+            )
+        else:
+            split_values = op.Split(
+                input, axis=axis, num_outputs=num_outputs, _outputs=split_outputs
+            )
     else:
         return None
 

{onnxscript-0.7.2.dev20260710 → onnxscript-0.7.2.dev20260714}/onnxscript/rewriter/rules/common/_basic_rules.py

@@ -125,7 +125,7 @@ class ReshapeReshape(RewriteRuleClassBase):
         return op.Reshape(op.Reshape(x, shape_ignored), shape)
 
     def rewrite(self, op, x: ir.Value, shape_ignored: ir.Value, shape: ir.Value):
-        new_shape = op.initializer(ir.Tensor(self._new_shape, name=shape.name))
+        new_shape = op.initializer(ir.Tensor(self._new_shape, name=self._new_shape_name))
         return op.Reshape(x, new_shape, allowzero=self._allowzero)
 
     def check(self, context, x, shape_ignored, shape) -> MatchResult:
@@ -145,6 +145,7 @@ class ReshapeReshape(RewriteRuleClassBase):
 
         # Constraints for shape.
         self._allowzero = context.nodes[0].attributes.get_int("allowzero", 0)
+        self._new_shape_name = f"{context.output_values[0].name}/shape"
         if self._allowzero == 1 and any(self._new_shape == 0):
             return check_result
         if any(self._new_shape == 0) and any(self._new_shape < 0):

{onnxscript-0.7.2.dev20260710 → onnxscript-0.7.2.dev20260714/onnxscript.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.2.dev20260710
+Version: 0.7.2.dev20260714
 Summary: Naturally author ONNX functions and models using a subset of Python
 Author-email: Microsoft Corporation <onnx@microsoft.com>
 License: MIT License
@@ -27,7 +27,7 @@ License: MIT License
         
 Project-URL: Homepage, https://microsoft.github.io/onnxscript/
 Project-URL: Repository, https://github.com/microsoft/onnxscript
-Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/e6fad72c99db6a5639ab8ee9bd9c04c9039c1cb0
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/d3489d49a9d63d039545e74c0ce80cbbbe241236
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers