Handle escaping addresses in LocalVariableAccessWalker

atrick · atrick · commit a7a6cc59d449 · 2024-03-26T11:03:05.000-07:00
Fixes rdar://124564951 (Compiler crash when evaluating pointer escape in autoclosure; LifetimeDepenenceScopeFixup; Fatal error: check findPointerEscape() before computing interior liveness.) (cherry picked from commit f1fbbcd)
diff --git a/SwiftCompilerSources/Sources/Optimizer/Utilities/LocalVariableUtils.swift b/SwiftCompilerSources/Sources/Optimizer/Utilities/LocalVariableUtils.swift
@@ -441,7 +441,8 @@ extension LocalVariableAccessWalker: AddressUseVisitor {
   }
 
   mutating func escapingAddressUse(of operand: Operand) -> WalkResult {
-    return .abortWalk
+    visit(LocalVariableAccess(.escape, operand))
+    return .continueWalk
   }
 
   mutating func unknownAddressUse(of operand: Operand) -> WalkResult {

Original file line number	Diff line number	Diff line change
`@@ -441,7 +441,8 @@ extension LocalVariableAccessWalker: AddressUseVisitor {`
`441`	`441`	`}`
`442`	`442`
`443`	`443`	`mutating func escapingAddressUse(of operand: Operand) -> WalkResult {`
`444`		`- return .abortWalk`
	`444`	`+ visit(LocalVariableAccess(.escape, operand))`
	`445`	`+ return .continueWalk`
`445`	`446`	`}`
`446`	`447`
`447`	`448`	`mutating func unknownAddressUse(of operand: Operand) -> WalkResult {`