[RISCV][GlobalISel] Generate integer load selection

1 files changed, 14 insertions, 0 deletions

diff --git a/backend/utils/TableGen/Common/CodeGenSelection.cpp b/backend/utils/TableGen/Common/CodeGenSelection.cpp
index defe98b..0d59d6b 100644
--- a/backend/utils/TableGen/Common/CodeGenSelection.cpp
+++ b/backend/utils/TableGen/Common/CodeGenSelection.cpp
@@ -297,6 +297,20 @@ bool CodeGenTarget::normalizeSelectionRules(NormalizedTargetModel &Result) {
       ++Normalized.Specificity;
     }
 
+    if (Rule.MemoryPredicate) {
+      if (Rule.MemoryPredicate->Bytes == 0 ||
+          (MatchInstruction != nullptr && !MatchInstruction->RequiresMemoryOperands)) {
+        fail(Rule.MemoryPredicate->Range,
+             "selection memory predicate requires a nonzero size and a memory opcode");
+        return false;
+      }
+      Normalized.MemoryPredicate = {
+          .MemoryOperandIndex = Rule.MemoryPredicate->MemoryOperandIndex,
+          .Bytes = Rule.MemoryPredicate->Bytes,
+      };
+      ++Normalized.Specificity;
+    }
+
     std::unordered_set<std::string> TargetPredicateSymbols;
     for (const SourceSelectionTargetPredicate &Predicate : Rule.TargetPredicates) {
       if (!isIdentifier(Predicate.Symbol) ||