Merge pull request #42026 from eeckstein/fix-release-devirtualizer

fix a bug in the ReleaseDevirtualizer pass

Author: eeckstein

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/ReleaseDevirtualizer.swift

@@ -49,10 +49,15 @@ let releaseDevirtualizerPass = FunctionPass(
           }
         }
 
-        if instruction is ReleaseValueInst || instruction is StrongReleaseInst {
-          lastRelease = instruction as? RefCountingInst
-        } else if instruction.mayRelease {
-          lastRelease = nil
+        switch instruction {
+          case is ReleaseValueInst, is StrongReleaseInst:
+            lastRelease = instruction as? RefCountingInst
+          case is DeallocRefInst, is SetDeallocatingInst:
+            lastRelease = nil
+          default:
+            if instruction.mayRelease {
+              lastRelease = nil
+            }
         }
       }
     }

test/SILOptimizer/devirt_release.sil

@@ -3,6 +3,8 @@
 // on the same platform at the same time.
 // RUN: %target-sil-opt -enable-sil-verify-all %s -release-devirtualizer -module-name=test | %FileCheck %s
 
+// REQUIRES: swift_in_compiler
+
 sil_stage canonical
 
 import Builtin
@@ -43,6 +45,27 @@ bb0:
   return %r : $()
 }
 
+// CHECK-LABEL: sil @dont_devirtualize_devirtualized
+// CHECK: alloc_ref
+// CHECK-NEXT: strong_retain
+// CHECK-NEXT: strong_release
+// CHECK-NEXT: set_deallocating
+// CHECK-NEXT: dealloc_ref
+// CHECK-NEXT: dealloc_stack_ref
+// CHECK-NEXT: tuple
+// CHECK: } // end sil function 'dont_devirtualize_devirtualized'
+sil @dont_devirtualize_devirtualized : $@convention(thin) () -> () {
+bb0:
+  %1 = alloc_ref [stack] $B
+  strong_retain %1 : $B
+  strong_release %1 : $B
+  set_deallocating %1 : $B
+  dealloc_ref %1 : $B
+  dealloc_stack_ref %1 : $B
+  %r = tuple ()
+  return %r : $()
+}
+
 // CHECK-LABEL: sil @dont_devirtualize_wrong_release
 // CHECK: alloc_ref
 // CHECK-NEXT: strong_release