From f3dbbeec9d32a98a2be4a8b13ffa78f65499d154 Mon Sep 17 00:00:00 2001
From: Owen Anderson <resistor@mac.com>
Date: Thu, 24 Jul 2025 12:13:59 +0800
Subject: [PATCH] [CHERIOT] Disallow more ways of forming negative offsets in
 LSR.

---
 .../Target/RISCV/RISCVTargetTransformInfo.cpp | 28 ++++++++++++-------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.cpp b/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.cpp
index da1a1103cf033..39631da1abd8b 100644
--- a/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.cpp
+++ b/llvm/lib/Target/RISCV/RISCVTargetTransformInfo.cpp
@@ -2404,20 +2404,28 @@ RISCVTTIImpl::getPreferredAddressingMode(const Loop *L,
 
 bool RISCVTTIImpl::isLegalBaseRegForLSR(const SCEV *S) const {
   if (ST->hasVendorXCheriot()) {
-    // Disallow any add-recurrence SCEV where the base offset is negative.
+    // Disallow any SCEV where the base offset is negative.
     // This is needed because CHERIoT can't represent pointers before the
     // beginning of an array.
+    auto IsAddOfNegativeCst = [](const SCEVAddExpr *A) {
+      if (!A)
+        return false;
+      const auto *Offset = dyn_cast<SCEVConstant>(A->getOperand(0));
+      if (Offset && Offset->getValue()->isNegative())
+        return true;
+      Offset = dyn_cast<SCEVConstant>(A->getOperand(1));
+      if (Offset && Offset->getValue()->isNegative())
+        return true;
+      return false;
+    };
+
     if (const auto *AddRec = dyn_cast<SCEVAddRecExpr>(S)) {
-      const auto *StartAdd = dyn_cast<SCEVAddExpr>(AddRec->getStart());
-      if (StartAdd) {
-        const auto *Offset = dyn_cast<SCEVConstant>(StartAdd->getOperand(0));
-        if (Offset && Offset->getValue()->isNegative())
-          return false;
-        Offset = dyn_cast<SCEVConstant>(StartAdd->getOperand(1));
-        if (Offset && Offset->getValue()->isNegative())
-          return false;
-      }
+      if (const auto *Cst = dyn_cast<SCEVConstant>(AddRec->getStart()))
+        return !Cst->getValue()->isNegative();
+      return !IsAddOfNegativeCst(dyn_cast<SCEVAddExpr>(AddRec->getStart()));
     }
+
+    return !IsAddOfNegativeCst(dyn_cast<SCEVAddExpr>(S));
   }
   return BasicTTIImplBase::isLegalBaseRegForLSR(S);
 }