diff --git a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
index f2a7220b4bed..12e8b257ce9f 100644
--- a/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
+++ b/mlir/lib/Dialect/Tensor/IR/TensorOps.cpp
@@ -1676,7 +1676,7 @@ public:
         auto flattenedIndex = sourceAttr.getFlattenedIndex(indices);
         std::vector<Attribute> updatedValues;
         updatedValues.reserve(sourceAttr.getNumElements());
-        for (auto i = 0; i < sourceAttr.getNumElements(); ++i) {
+        for (unsigned i = 0; i < sourceAttr.getNumElements(); ++i) {
           updatedValues.push_back(i == flattenedIndex ? scalarAttr
                                                       : sourceValues[i]);
         }