[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