Add AccessedStorage::isGuaranteedForFunction

to quickly bypass all liveness checking in many cases.
yim-lee · Sep 21, 2021 · 19cff2e · 19cff2e
1 parent b750c37
commit 19cff2e
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/include/swift/SIL/MemAccessUtils.h b/include/swift/SIL/MemAccessUtils.h
@@ -591,6 +591,10 @@ class AccessedStorage {
     return getKind() == Class;
   }
 
+  /// Return true if this storage is valid for all uses in a function without
+  /// checking its lifetime.
+  bool isGuaranteedForFunction() const;
+
   /// Returns the ValueDecl for the underlying storage, if it can be
   /// determined. Otherwise returns null.
   ///

diff --git a/lib/SIL/Utils/MemAccessUtils.cpp b/lib/SIL/Utils/MemAccessUtils.cpp
@@ -570,6 +570,17 @@ void AccessedStorage::visitRoots(
   }
 }
 
+bool AccessedStorage::isGuaranteedForFunction() const {
+  if (getKind() == AccessedStorage::Argument) {
+    return getArgument()->getArgumentConvention().isGuaranteedConvention();
+  }
+  if (isObjectAccess()) {
+    return getRoot().getOwnershipKind() == OwnershipKind::Guaranteed
+           && isa<SILFunctionArgument>(getRoot());
+  }
+  return false;
+}
+
 // Set 'isLet' to true if this storage can be determined to be a 'let' variable.
 //
 // \p base must be the access base for this storage, as passed to the