From dc63ca78f740b61c92f918fbb172ecf632f7e8a4 Mon Sep 17 00:00:00 2001
From: Jeff Niu <jeff@modular.com>
Date: Fri, 21 Oct 2022 13:13:07 -0700
Subject: [PATCH] [mlir][sparse] Fix build warning (NFC)

---
 mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
index b793495..22282ca 100644
--- a/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
+++ b/mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp
@@ -626,7 +626,7 @@ struct OutRewriter : public OpRewritePattern<OutOp> {
     // the buffer.
     SmallVector<Value, 4> dims;
     sizesForTensor(rewriter, dims, loc, srcTp, src);
-    for (int64_t i = 0; i < rank; i++) {
+    for (uint64_t i = 0; i < rank; i++) {
       rewriter.create<memref::StoreOp>(loc, dims[i], dimSizes,
                                        constantIndex(rewriter, loc, i));
     }
@@ -650,7 +650,7 @@ struct OutRewriter : public OpRewritePattern<OutOp> {
     // For each element in the source tensor, output the element.
     rewriter.create<ForeachOp>(
         loc, src, [&](OpBuilder &builder, Location loc, ValueRange args) {
-          for (int64_t i = 0; i < rank; i++) {
+          for (uint64_t i = 0; i < rank; i++) {
             rewriter.create<memref::StoreOp>(loc, args[i], indices,
                                              constantIndex(builder, loc, i));
           }
-- 
2.7.4