diff options
Diffstat (limited to 'mlir/lib/Rewrite/ByteCode.cpp')
| -rw-r--r-- | mlir/lib/Rewrite/ByteCode.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/mlir/lib/Rewrite/ByteCode.cpp b/mlir/lib/Rewrite/ByteCode.cpp index d4ca8b2..388d6dc 100644 --- a/mlir/lib/Rewrite/ByteCode.cpp +++ b/mlir/lib/Rewrite/ByteCode.cpp @@ -1713,11 +1713,11 @@ executeGetOperandsResults(RangeT values, Operation *op, unsigned index, LLVM_DEBUG(llvm::dbgs() << " * Extracting values from `" << attrSizedSegments << "`\n"); - auto segmentAttr = op->getAttrOfType<DenseElementsAttr>(attrSizedSegments); - if (!segmentAttr || segmentAttr.getNumElements() <= index) + auto segmentAttr = op->getAttrOfType<DenseI32ArrayAttr>(attrSizedSegments); + if (!segmentAttr || segmentAttr.asArrayRef().size() <= index) return nullptr; - auto segments = segmentAttr.getValues<int32_t>(); + ArrayRef<int32_t> segments = segmentAttr; unsigned startIndex = std::accumulate(segments.begin(), segments.begin() + index, 0); values = values.slice(startIndex, *std::next(segments.begin(), index)); |