onnxscript 0.7.2.dev20260814__tar.gz → 0.7.2.dev20260821__tar.gz

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

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

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

@@ -4189,6 +4189,7 @@ def aten_full_like(
     layout: str = "",
     device: str = "",
     pin_memory: bool = False,
+    memory_format: str = "",
 ) -> TensorType:
     """full_like(Tensor self, Scalar fill_value, *, ScalarType? dtype=None, Layout? layout=None, Device? device=None, bool? pin_memory=None, MemoryFormat? memory_format=None) -> Tensor"""
 

{onnxscript-0.7.2.dev20260814 → onnxscript-0.7.2.dev20260821}/onnxscript/function_libs/torch_lib/ops/fft.py

@@ -172,17 +172,33 @@ def aten__fft_r2c(
         transformed = op.Unsqueeze(self, axes=[-1])
 
     for idx, dimension in enumerate(reversed(dim)):
+        dft_length = op.Squeeze(
+            op.Shape(transformed, start=dimension, end=dimension + 1),
+            axes=[0],
+        )
         transformed = _fftn_onnx_normalization(
             transformed,
             normalization,
-            op.Shape(transformed, start=dimension, end=dimension + 1),
+            dft_length,
             inverse=False,
         )
         if idx > 0:
-            transformed = op.DFT(transformed, axis=dimension, inverse=False, onesided=False)
+            transformed = op.DFT(
+                transformed,
+                dft_length=dft_length,
+                axis=dimension,
+                inverse=False,
+                onesided=False,
+            )
         else:
             # Torch computes one-sided FFT on the last dimension only.
-            transformed = op.DFT(transformed, axis=dimension, inverse=False, onesided=onesided)
+            transformed = op.DFT(
+                transformed,
+                dft_length=dft_length,
+                axis=dimension,
+                inverse=False,
+                onesided=onesided,
+            )
 
     if unsqueeze_first_dim:
         transformed = op.Squeeze(transformed, axes=[0])

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

@@ -1713,13 +1713,17 @@ def _causal_attention_mask(query: TFloat, key: TFloat) -> TFloat:
     attn_mask = op.Expand(op.Constant(value_float=1.0), size)
     # }
     attn_mask = op.Trilu(attn_mask, upper=0)
-    # The causal mask has 0s in the lower triangle and -inf in the upper triangle.
+    # The causal mask has 0s in the lower triangle and the lowest representable value
+    # of the query dtype in the upper triangle. The lowest finite value is used instead
+    # of -inf to stay consistent with the boolean mask path and to avoid producing NaN
+    # in the subsequent Softmax when a row is fully masked.
+    # The constants are built in the query dtype directly rather than in float32 and
+    # cast down, because casting float32's lowest value to float16 overflows to -inf.
     attn_mask = op.Where(
         op.Equal(attn_mask, op.Constant(value_float=0.0)),
-        op.Constant(value_float=-float("inf")),
-        op.Constant(value_float=0.0),
+        op.Constant(value=ir.tensor(query.dtype.min, dtype=query.dtype)),
+        op.Constant(value=ir.tensor(0.0, dtype=query.dtype)),
     )
-    attn_mask = op.CastLike(attn_mask, query)
     return attn_mask
 
 

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

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