================
@@ -32,6 +33,68 @@ namespace mlir {
namespace {
+/// Find the `cir.store` operation that stores to the given alloca and
dominates
+/// the given load operation. Dominance calculation is done through the given
+/// DominanceInfo object.
+///
+/// Return nullptr if no such store operation exists. If multiple store
+/// operations satisfy the criteria, any of them may be returned.
+cir::StoreOp findDominatingInitOp(cir::AllocaOp alloca, cir::LoadOp load,
+ const DominanceInfo &domInfo) {
+ // Walk through all uses of the alloca and visit the store operations that
+ // store to the alloca
+ for (const mlir::OpOperand &use : alloca->getUses()) {
+ auto store = mlir::dyn_cast<cir::StoreOp>(use.getOwner());
+ if (!store)
+ continue;
+
+ // `cir.store` has two operands, we're only interested if the store is
+ // storing into the alloca, not if the store is storing the address of the
+ // alloca slot into somewhere else
+ if (use.getOperandNumber() != cir::StoreOp::odsIndex_addr)
----------------
Lancern wrote:
Updated to give up optimization if the store found is either volatile or atomic.
https://github.com/llvm/llvm-project/pull/212284
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
