Merge pull request #22519 from gottesmm/pr-bafb507409a233c2b64c95d66aa152e937fe7b86

[gardening] Change an if-else over ApplySite to a covered switch over…

Author: gottesmm

lib/SILOptimizer/IPO/UsePrespecialized.cpp

@@ -55,6 +55,8 @@ class UsePrespecialized: public SILModuleTransform {
   bool replaceByPrespecialized(SILFunction &F);
 };
 
+} // end anonymous namespace
+
 // Analyze the function and replace each apply of
 // a generic function by an apply of the corresponding
 // pre-specialized function, if such a pre-specialization exists.
@@ -143,12 +145,17 @@ bool UsePrespecialized::replaceByPrespecialized(SILFunction &F) {
                             << " : " << NewF->getName() << "\n");
 
     auto NewAI = replaceWithSpecializedFunction(AI, NewF, ReInfo);
-    if (auto oldApply = dyn_cast<ApplyInst>(AI)) {
-      oldApply->replaceAllUsesWith(cast<ApplyInst>(NewAI));
-    } else if (auto oldPApply = dyn_cast<PartialApplyInst>(AI)) {
-      oldPApply->replaceAllUsesWith(cast<PartialApplyInst>(NewAI));
-    } else {
-      assert(isa<TryApplyInst>(NewAI) || isa<BeginApplyInst>(NewAI));
+    switch (AI.getKind()) {
+    case ApplySiteKind::ApplyInst:
+      cast<ApplyInst>(AI)->replaceAllUsesWith(cast<ApplyInst>(NewAI));
+      break;
+    case ApplySiteKind::PartialApplyInst:
+      cast<PartialApplyInst>(AI)->replaceAllUsesWith(
+          cast<PartialApplyInst>(NewAI));
+      break;
+    case ApplySiteKind::TryApplyInst:
+    case ApplySiteKind::BeginApplyInst:
+      break;
     }
     recursivelyDeleteTriviallyDeadInstructions(AI.getInstruction(), true);
     Changed = true;
@@ -157,8 +164,6 @@ bool UsePrespecialized::replaceByPrespecialized(SILFunction &F) {
   return Changed;
 }
 
-} // end anonymous namespace
-
 
 SILTransform *swift::createUsePrespecialized() {
   return new UsePrespecialized();