[mlir] [MLIR][Linalg] Introduce broadcast/transpose semantic to 'linalg.batc… (PR #122275)
Andrzej Warzyński
llvmlistbot at llvm.org
Wed Jan 15 11:53:08 PST 2025
================
@@ -88,6 +88,14 @@ FailureOr<Operation *>
mlir::linalg::transposeBatchMatmul(RewriterBase &rewriter,
linalg::BatchMatmulOp batchMatmulOp,
bool transposeLHS) {
+ // Check to not let go the batch_matmul with extended semantic, through this
+ // transform.
+ if (batchMatmulOp.hasUserDefinedMaps()) {
+ return rewriter.notifyMatchFailure(
+ batchMatmulOp,
+ "only batch_matmul ops with non-extended semantics are supported");
----------------
banach-space wrote:
Either `linalg.batch_matmul` or skip altogether. IIRC, the Op will be printed anyway. Also:
```suggestion
"user-defined maps are not supported");
```
https://github.com/llvm/llvm-project/pull/122275
More information about the Mlir-commits
mailing list