[Mlir-commits] [mlir] [MLIR][Linalg] Expose linalg.matmul and linalg.contract via Python API (PR #126377)

Rolf Morel llvmlistbot at llvm.org
Sun Feb 9 10:42:54 PST 2025


================
@@ -147,3 +147,51 @@ def __init__(
 
 
 generic = region_op(GenericOp_, terminator=YieldOp)
+
+
+def matmul(
+    inputs: Sequence[Union[Operation, OpView, Value]],
+    *,
+    outs: Sequence[Union[Operation, OpView, Value]],
+    indexing_maps: Sequence[AffineMapAttr],
----------------
rolfmorel wrote:

Was just a copy-paste oversight - I started with ContractOp. Only when this was remedied did all the details of the arcane way to construct ops (from Python) start to matter. 

https://github.com/llvm/llvm-project/pull/126377


More information about the Mlir-commits mailing list