Addressed upstream comment

Change-Id: I02495c6cf2d67fcf7e0d266c8b51c661e759a3f4

Author: d-smirnov

mlir/lib/Dialect/MemRef/Utils/MemRefUtils.cpp

@@ -141,7 +141,7 @@ static bool resultIsNotRead(Operation *op, std::vector<Operation *> &uses) {
   for (OpOperand &use : op->getUses()) {
     Operation *useOp = use.getOwner();
     // Use escaped the scope
-    if (useOp->hasTrait<mlir::OpTrait::ReturnLike>())
+    if (useOp->mightHaveTrait<OpTrait::IsTerminator>())
       return false;
     if (isa<memref::DeallocOp>(useOp) ||
         (useOp->getNumResults() == 0 && useOp->getNumRegions() == 0 &&