Restore range check for memory opcodes, too

Author: s-barannikov

llvm/lib/Target/Mips/MipsISelLowering.h

@@ -247,14 +247,16 @@ class TargetRegisterClass;
       DOUBLE_SELECT_I64,
 
       // Load/Store Left/Right nodes.
-      LWL,
+      FIRST_MEMORY_OPCODE,
+      LWL = FIRST_MEMORY_OPCODE,
       LWR,
       SWL,
       SWR,
       LDL,
       LDR,
       SDL,
-      SDR
+      SDR,
+      LAST_MEMORY_OPCODE = SDR,
     };
 
   } // ene namespace MipsISD

llvm/lib/Target/Mips/MipsSelectionDAGInfo.cpp

@@ -14,17 +14,6 @@ using namespace llvm;
 MipsSelectionDAGInfo::~MipsSelectionDAGInfo() = default;
 
 bool MipsSelectionDAGInfo::isTargetMemoryOpcode(unsigned Opcode) const {
-  switch (static_cast<MipsISD::NodeType>(Opcode)) {
-  default:
-    return false;
-  case MipsISD::LWL:
-  case MipsISD::LWR:
-  case MipsISD::SWL:
-  case MipsISD::SWR:
-  case MipsISD::LDL:
-  case MipsISD::LDR:
-  case MipsISD::SDL:
-  case MipsISD::SDR:
-    return true;
-  }
+  return Opcode >= MipsISD::FIRST_MEMORY_OPCODE &&
+         Opcode <= MipsISD::LAST_MEMORY_OPCODE;
 }

llvm/lib/Target/RISCV/RISCVISelLowering.h

@@ -492,11 +492,13 @@ enum NodeType : unsigned {
   STRICT_VFROUND_NOEXCEPT_VL,
   LAST_STRICTFP_OPCODE = STRICT_VFROUND_NOEXCEPT_VL,
 
-  TH_LWD,
+  FIRST_MEMORY_OPCODE,
+  TH_LWD = FIRST_MEMORY_OPCODE,
   TH_LWUD,
   TH_LDD,
   TH_SWD,
   TH_SDD,
+  LAST_MEMORY_OPCODE = TH_SDD,
 };
 // clang-format on
 } // namespace RISCVISD

llvm/lib/Target/RISCV/RISCVSelectionDAGInfo.cpp

@@ -14,16 +14,8 @@ using namespace llvm;
 RISCVSelectionDAGInfo::~RISCVSelectionDAGInfo() = default;
 
 bool RISCVSelectionDAGInfo::isTargetMemoryOpcode(unsigned Opcode) const {
-  switch (static_cast<RISCVISD::NodeType>(Opcode)) {
-  default:
-    return false;
-  case RISCVISD::TH_LWD:
-  case RISCVISD::TH_LWUD:
-  case RISCVISD::TH_LDD:
-  case RISCVISD::TH_SWD:
-  case RISCVISD::TH_SDD:
-    return true;
-  }
+  return Opcode >= RISCVISD::FIRST_MEMORY_OPCODE &&
+         Opcode <= RISCVISD::LAST_MEMORY_OPCODE;
 }
 
 bool RISCVSelectionDAGInfo::isTargetStrictFPOpcode(unsigned Opcode) const {