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

Maksim Levental llvmlistbot at llvm.org
Sat Feb 8 10:23:16 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],
----------------
makslevental wrote:

So my biggest confusion with the previous changes and the newly introduced requirements is: why do we suddenly need the user to provide indexing maps? I mean the indexing patterns for matmul are of course fixed and haven't changed so is it because we simply don't have implementations in C++ of those opdsl helpers that were prior computing these indexing maps in Python?

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


More information about the Mlir-commits mailing list