diff --git a/mlir/lib/IR/StandardTypes.cpp b/mlir/lib/IR/StandardTypes.cpp --- a/mlir/lib/IR/StandardTypes.cpp +++ b/mlir/lib/IR/StandardTypes.cpp @@ -199,7 +199,9 @@ int64_t ShapedType::getRank() const { return getShape().size(); } -bool ShapedType::hasRank() const { return !isa(); } +bool ShapedType::hasRank() const { + return !isa() && !isa(); +} int64_t ShapedType::getDimSize(unsigned idx) const { assert(idx < getRank() && "invalid index for shaped type");