Encode bitwidth once for ConstantRangeList kind attr

llvm · May 9, 2024 · 8e64fa9 · 8e64fa9
1 parent e52fd9d
commit 8e64fa9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/llvm/lib/Bitcode/Reader/BitcodeReader.cpp b/llvm/lib/Bitcode/Reader/BitcodeReader.cpp
@@ -2344,13 +2344,14 @@ Error BitcodeReader::parseAttributeGroupBlock() {
 
           SmallVector<ConstantRange, 2> Val;
           unsigned RangeSize = Record[++i];
+          unsigned BitWidth = Record[++i];
           if (i + 2 * RangeSize >= e)
             return error("Incomplete constant range list");
           for (unsigned Idx = 0; Idx < RangeSize; ++Idx) {
             int64_t Start = BitcodeReader::decodeSignRotatedValue(Record[++i]);
             int64_t End = BitcodeReader::decodeSignRotatedValue(Record[++i]);
-            Val.push_back(
-                ConstantRange(APInt(64, Start, true), APInt(64, End, true)));
+            Val.push_back(ConstantRange(APInt(BitWidth, Start, true),
+                                        APInt(BitWidth, End, true)));
           }
           B.addConstantRangeListAttr(Kind, Val);
         } else {

diff --git a/llvm/lib/Bitcode/Writer/BitcodeWriter.cpp b/llvm/lib/Bitcode/Writer/BitcodeWriter.cpp
@@ -958,6 +958,7 @@ void ModuleBitcodeWriter::writeAttributeGroupTable() {
         Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum()));
         ArrayRef<ConstantRange> Val = Attr.getValueAsConstantRangeList();
         Record.push_back(Val.size());
+        Record.push_back(Val[0].getBitWidth());
         for (auto &CR : Val) {
           emitConstantRange(Record, CR, /*EmitBitWidth=*/false);
         }