Merge pull request #26609 from gottesmm/pr-cfce6f85ef30229ea80e5e1a659360d5e15f9fd1

Author: swift-ci

lib/SILOptimizer/Utils/ConstantFolding.cpp

@@ -1746,6 +1746,25 @@ ConstantFolder::processWorkList() {
       continue;
     }
 
+    // See if we have a CondFailMessage that we can canonicalize.
+    if (isApplyOfBuiltin(*I, BuiltinValueKind::CondFailMessage)) {
+      // See if our operand is a string literal inst. In such a case, fold into
+      // cond_fail instruction.
+      if (auto *sli = dyn_cast<StringLiteralInst>(I->getOperand(1))) {
+        if (sli->getEncoding() == StringLiteralInst::Encoding::UTF8) {
+          SILBuilderWithScope builder(I);
+          auto *cfi = builder.createCondFail(I->getLoc(), I->getOperand(0),
+                                             sli->getValue());
+          WorkList.insert(cfi);
+          recursivelyDeleteTriviallyDeadInstructions(
+              I, /*force*/ true,
+              [&](SILInstruction *DeadI) { WorkList.remove(DeadI); });
+          InvalidateInstructions = true;
+        }
+      }
+      continue;
+    }
+
     // Go through all users of the constant and try to fold them.
     FoldedUsers.clear();
     for (auto Result : I->getResults()) {
@@ -1784,29 +1803,12 @@ ConstantFolder::processWorkList() {
           WorkList.insert(User);
         }
 
-        // See if we have a CondFailMessage. If we do, see if we can transform
-        // it into a UTF8.
-        if (auto *bi = dyn_cast<BuiltinInst>(User)) {
-          if (auto kind = bi->getBuiltinKind()) {
-            if (*kind == BuiltinValueKind::CondFailMessage) {
-              // See if our original instruction was a string literal inst.
-              if (auto *sli = dyn_cast<StringLiteralInst>(I)) {
-                if (sli->getEncoding() == StringLiteralInst::Encoding::UTF8) {
-                  SILBuilderWithScope builder(bi);
-                  auto *cfi = builder.createCondFail(
-                      bi->getLoc(), bi->getOperand(0), sli->getValue());
-                  WorkList.insert(cfi);
-                  recursivelyDeleteTriviallyDeadInstructions(
-                      bi, /*force*/ true,
-                      [&](SILInstruction *DeadI) { WorkList.remove(DeadI); });
-                  InvalidateInstructions = true;
-                  continue;
-                }
-              }
-
-              // If we weren't able to simplify into a cond_fail, add it to the
-              // folded user set to see if the condfail msg is dead.
-              FoldedUsers.insert(bi);
+        // See if we have a CondFailMessage of a string_Literal. If we do, add
+        // it to the worklist, so we can clean it up.
+        if (isApplyOfBuiltin(*User, BuiltinValueKind::CondFailMessage)) {
+          if (auto *sli = dyn_cast<StringLiteralInst>(I)) {
+            if (sli->getEncoding() == StringLiteralInst::Encoding::UTF8) {
+              WorkList.insert(User);
             }
           }
         }