Track memops more accurately (#43778)

Author: pchintalapudi

src/llvm-alloc-helpers.cpp

@@ -80,8 +80,6 @@ bool AllocUseInfo::addMemOp(Instruction *inst, unsigned opno, uint32_t offset,
     MemOp memop(inst, opno);
     memop.offset = offset;
     uint64_t size = DL.getTypeStoreSize(elty);
-    if (size >= UINT32_MAX - offset)
-        return false;
     memop.size = size;
     memop.isaggr = isa<StructType>(elty) || isa<ArrayType>(elty) || isa<VectorType>(elty);
     memop.isobjref = hasObjref(elty);
@@ -105,6 +103,8 @@ bool AllocUseInfo::addMemOp(Instruction *inst, unsigned opno, uint32_t offset,
         field.second.hasaggr = true;
     }
     field.second.accesses.push_back(memop);
+    if (size >= UINT32_MAX - offset)
+        return false;
     return true;
 }
 

src/llvm-alloc-helpers.h

@@ -27,8 +27,8 @@ namespace jl_alloc {
 
     struct MemOp {
         llvm::Instruction *inst;
+        uint64_t offset = 0;
         unsigned opno;
-        uint32_t offset = 0;
         uint32_t size = 0;
         bool isobjref:1;
         bool isaggr:1;