diff options
| -rw-r--r-- | mlir/include/mlir/IR/Attributes.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/mlir/include/mlir/IR/Attributes.h b/mlir/include/mlir/IR/Attributes.h index 49f42affb56..ed82ef11175 100644 --- a/mlir/include/mlir/IR/Attributes.h +++ b/mlir/include/mlir/IR/Attributes.h @@ -220,10 +220,11 @@ private: /// Class for underlying value iterator support. template <typename AttrTy> class attr_value_iterator final - : public llvm::mapped_iterator<iterator, AttrTy (*)(Attribute)> { + : public llvm::mapped_iterator<ArrayAttr::iterator, + AttrTy (*)(Attribute)> { public: - explicit attr_value_iterator(iterator it) - : llvm::mapped_iterator<iterator, AttrTy (*)(Attribute)>( + explicit attr_value_iterator(ArrayAttr::iterator it) + : llvm::mapped_iterator<ArrayAttr::iterator, AttrTy (*)(Attribute)>( it, [](Attribute attr) { return attr.cast<AttrTy>(); }) {} AttrTy operator*() { return (*this->I).template cast<AttrTy>(); } }; |
