fixup! fixup! [mlir][linalg] Prevent hoisting of transfer pairs in the presence of aliases

Make sure only uses inside the loop are counted

Author: banach-space

mlir/lib/Dialect/Linalg/Transforms/Hoisting.cpp

@@ -330,7 +330,12 @@ void mlir::linalg::hoistRedundantVectorTransfers(Operation *root,
         //  `memref.assume_alignment`)
         if (auto assume = dyn_cast<memref::AssumeAlignmentOp>(source)) {
           Value memPreAlignment = assume.getMemref();
-          if (base.hasNUses(2) && memPreAlignment.hasOneUse())
+          auto numInLoopUses =
+              llvm::count_if(base.getUses(), [&loop](OpOperand &use) {
+                return loop->isAncestor(use.getOwner());
+              });
+
+          if (numInLoopUses && memPreAlignment.hasOneUse())
             source = memPreAlignment.getDefiningOp();
         }
         if (isa_and_nonnull<ViewLikeOpInterface>(source))