diff --git a/mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp b/mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp
--- a/mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp
+++ b/mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp
@@ -761,7 +761,8 @@
 
   // Early exit for the case where the number of unused dims matches the number
   // of ranks reduced.
-  if (unusedDims.count() + reducedType.getRank() == originalType.getRank())
+  if (static_cast<int64_t>(unusedDims.count()) + reducedType.getRank() ==
+      originalType.getRank())
     return unusedDims;
 
   SmallVector<int64_t> originalStrides, candidateStrides;