1 files changed, 15 insertions, 10 deletions
diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyPeephole.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyPeephole.cpp
index 39a4bf9dce5..5f457f0a06e 100644
--- a/llvm/lib/Target/WebAssembly/WebAssemblyPeephole.cpp
+++ b/llvm/lib/Target/WebAssembly/WebAssemblyPeephole.cpp
@@ -113,16 +113,21 @@ bool WebAssemblyPeephole::runOnMachineFunction(MachineFunction &MF) {
               Name == TLI.getLibcallName(RTLIB::MEMSET)) {
             LibFunc::Func Func;
             if (LibInfo.getLibFunc(Name, Func)) {
-              if (!MI.getOperand(2).isReg())
-                report_fatal_error(
-                    "Call to builtin function with wrong signature");
-              MachineOperand &MO = MI.getOperand(0);
-              unsigned OldReg = MO.getReg();
-              unsigned NewReg = MI.getOperand(2).getReg();
-              if (MRI.getRegClass(NewReg) != MRI.getRegClass(OldReg))
-                report_fatal_error(
-                    "Call to builtin function with wrong signature");
-              Changed |= MaybeRewriteToDiscard(OldReg, NewReg, MO, MFI, MRI);
+              const auto &Op2 = MI.getOperand(2);
+              if (Op2.isReg()) {
+                MachineOperand &MO = MI.getOperand(0);
+                unsigned OldReg = MO.getReg();
+                unsigned NewReg = Op2.getReg();
+                if (MRI.getRegClass(NewReg) != MRI.getRegClass(OldReg))
+                  report_fatal_error("Peephole: call to builtin function with "
+                                     "wrong signature, from/to mismatch");
+                Changed |= MaybeRewriteToDiscard(OldReg, NewReg, MO, MFI, MRI);
+              } else if (Op2.isFI()) {
+                break;
+              } else {
+                report_fatal_error("Peephole: call to builtin function with "
+                                   "wrong signature, not consuming reg");
+              }
             }
           }
         }