Make _resultDependsOnSelf available on SILFunction

Author: meg-gupta

include/swift/SIL/SILArgument.h

@@ -366,8 +366,8 @@ class SILFunctionArgument : public SILArgument {
       ValueOwnershipKind ownershipKind, const ValueDecl *decl = nullptr,
       bool isNoImplicitCopy = false,
       LifetimeAnnotation lifetimeAnnotation = LifetimeAnnotation::None,
-      bool isCapture = false,
-      bool isParameterPack = false, bool hasResultDependsOn = false)
+      bool isCapture = false, bool isParameterPack = false,
+      bool hasResultDependsOn = false)
       : SILArgument(ValueKind::SILFunctionArgument, parentBlock, type,
                     ownershipKind, decl) {
     sharedUInt32().SILFunctionArgument.noImplicitCopy = isNoImplicitCopy;

include/swift/SIL/SILFunction.h

@@ -443,6 +443,8 @@ class SILFunction
   /// lifetime-dependence on an argument.
   unsigned HasUnsafeNonEscapableResult : 1;
 
+  unsigned HasResultDependsOnSelf : 1;
+
   static void
   validateSubclassScope(SubclassScope scope, IsThunk_t isThunk,
                         const GenericSpecializationInformation *genericInfo) {
@@ -727,6 +729,11 @@ class SILFunction
     HasUnsafeNonEscapableResult = value;
   }
 
+  bool hasResultDependsOnSelf() const { return HasResultDependsOnSelf; }
+
+  void setHasResultDependsOnSelf(bool flag = true) {
+    HasResultDependsOnSelf = flag;
+  }
   /// Returns true if this is a reabstraction thunk of escaping function type
   /// whose single argument is a potentially non-escaping closure. i.e. the
   /// thunks' function argument may itself have @inout_aliasable parameters.

lib/SIL/IR/SILFunction.cpp

@@ -209,6 +209,7 @@ void SILFunction::init(
   this->ForceEnableLexicalLifetimes = DoNotForceEnableLexicalLifetimes;
   this->UseStackForPackMetadata = DoUseStackForPackMetadata;
   this->HasUnsafeNonEscapableResult = false;
+  this->HasResultDependsOnSelf = false;
   this->stackProtection = false;
   this->Inlined = false;
   this->Zombie = false;

lib/SIL/IR/SILFunctionBuilder.cpp

@@ -209,6 +209,10 @@ void SILFunctionBuilder::addFunctionAttributes(
     F->setHasUnsafeNonEscapableResult(true);
   }
 
+  if (Attrs.hasAttribute<ResultDependsOnSelfAttr>()) {
+    F->setHasResultDependsOnSelf();
+  }
+
   // Validate `@differentiable` attributes by calling `getParameterIndices`.
   // This is important for:
   // - Skipping invalid `@differentiable` attributes in non-primary files.

lib/SIL/IR/SILPrinter.cpp

@@ -3416,6 +3416,10 @@ void SILFunction::print(SILPrintContext &PrintCtx) const {
   if (needsStackProtection())
     OS << "[stack_protection] ";
 
+  if (hasResultDependsOnSelf()) {
+    OS << "[_resultDependsOnSelf] ";
+  }
+
   llvm::DenseMap<CanType, Identifier> sugaredTypeNames;
   printSILFunctionNameAndType(OS, this, sugaredTypeNames, &PrintCtx);