onnxscript 0.7.2.dev20260805__tar.gz → 0.7.2.dev20260814__tar.gz

{onnxscript-0.7.2.dev20260805/onnxscript.egg-info → onnxscript-0.7.2.dev20260814}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.2.dev20260805
+Version: 0.7.2.dev20260814
 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/457a26066515ea85f719f2daf0e3cd853a015404
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/ab2b5f42395e2e6f4e32237885b82e06244e97f7
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers

{onnxscript-0.7.2.dev20260805 → onnxscript-0.7.2.dev20260814}/onnxscript/function_libs/torch_lib/ops/core.py

@@ -4094,8 +4094,12 @@ def aten_floor_divide(self: TTensor, other: TTensor) -> TTensor:
     # Reference: https://github.com/pytorch/pytorch/blob/ffc645c870f0abd368606ba1e2b3b58cacb03046/torch/_refs/__init__.py#L1401C1-L1409C70
     # offset = (torch.signbit(a) != torch.signbit(b)).logical_and(torch.fmod(a, b) != 0)
     # return prims.div(a, b) - _maybe_convert_to_dtype(offset, a.dtype)
+    # The sign-mismatch check ``signbit(a) != signbit(b)`` (for nonzero divisors)
+    # is expressed as ``(a < 0) == (b > 0)``, which avoids Sign (limited integral
+    # support on some EPs) and enables optimizations when ``self`` is provably
+    # non-negative.
     offset = op.And(
-        op.Not(op.Equal(op.Sign(self), op.Sign(other))),
+        op.Equal(op.Less(self, 0), op.Greater(other, 0)),
         op.Cast(op.Mod(self, other), to=BOOL.dtype),
     )
     offset = op.Cast(offset, to=self.dtype)

{onnxscript-0.7.2.dev20260805 → onnxscript-0.7.2.dev20260814}/onnxscript/function_libs/torch_lib/ops/nn.py

@@ -822,11 +822,17 @@ def aten_linear(input: TFloat, weight: TFloat, bias: Optional[TFloat] = None) ->
         # Use Gemm for the rank 2 input
         return op.Gemm(input, weight, bias, transB=True)
     if len(weight.shape) == 1:
-        # In rare cases the weight can be 1d
+        # In rare cases the weight can be 1d. aten::linear does not support a
+        # bias with a 1d weight (mat2 must be a matrix). eager linear contracts
+        # the last dim away in this case, so squeeze the dim added for MatMul
+        # back off.
+        if bias is not None:
+            raise NotImplementedError("aten::linear with 1d weight and bias is not supported")
+
         weight_transposed = op.Unsqueeze(weight, [1])
-    else:
-        assert len(weight.shape) == 2
-        weight_transposed = op.Transpose(weight, perm=[1, 0])
+        return op.Squeeze(op.MatMul(input, weight_transposed), [-1])
+    assert len(weight.shape) == 2
+    weight_transposed = op.Transpose(weight, perm=[1, 0])
     mul = op.MatMul(input, weight_transposed)
     if bias is None:
         return mul

{onnxscript-0.7.2.dev20260805 → onnxscript-0.7.2.dev20260814/onnxscript.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxscript
-Version: 0.7.2.dev20260805
+Version: 0.7.2.dev20260814
 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/457a26066515ea85f719f2daf0e3cd853a015404
+Project-URL: Commit, https://github.com/microsoft/onnxscript/tree/ab2b5f42395e2e6f4e32237885b82e06244e97f7
 Classifier: Development Status :: 4 - Beta
 Classifier: Environment :: Console
 Classifier: Intended Audience :: Developers