[Mlir-commits] [mlir] [mlir][x86] Add extra patterns for AMX nano-kernel lowering (acc with zero constant) (PR #222513)

[llvmlistbot at llvm.org]

================
@@ -1451,81 +1476,97 @@ struct VectorContractToAMXDotProduct
         k++;
       }
     }
+
+    // Update insertion point, if acc memref after contraction loop.
+    Operation *memrefOp = srcBuffAcc.getDefiningOp();
+    if (memrefOp && outerLoop &&
+        !(memrefOp->isBeforeInBlock(outerLoop.getOperation()))) {
+      rewriter.setInsertionPoint(memrefOp);
----------------
z6jh89mtvm-alt wrote:

S FX shaded

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