[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