onnxscript 0.7.2.dev20260903__tar.gz → 0.7.2.dev20260904__tar.gz

{onnxscript-0.7.2.dev20260903/onnxscript.egg-info → onnxscript-0.7.2.dev20260904}/PKG-INFO

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

{onnxscript-0.7.2.dev20260903 → onnxscript-0.7.2.dev20260904}/onnxscript/function_libs/torch_lib/ops/core.py

@@ -4673,6 +4673,27 @@ def aten_grid_sampler_3d_backward(
     raise NotImplementedError()
 
 
+@torch_op("aten::_grouped_mm", trace_only=True)
+def aten_grouped_mm(
+    self: TFloat,
+    mat2: TFloat,
+    offs: Optional[TInt] = None,
+    bias: Optional[TFloat] = None,
+    out_dtype: int = -1,
+) -> TFloat:
+    """_grouped_mm(Tensor self, Tensor mat2, *, Tensor? offs=None, Tensor? bias=None, int? out_dtype=None) -> Tensor"""
+
+    if offs is not None:
+        raise NotImplementedError("Grouped matmul with offsets (ragged/MoE) is not supported.")
+
+    res = op.MatMul(self, mat2)
+    if bias is not None:
+        res = op.Add(res, bias)
+    if out_dtype is not None and out_dtype != -1:
+        res = op.Cast(res, to=out_dtype)
+    return res
+
+
 def aten_gru_cell(
     input: TensorType,
     hx: TensorType,

{onnxscript-0.7.2.dev20260903 → onnxscript-0.7.2.dev20260904/onnxscript.egg-info}/PKG-INFO

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