From f9ad189c14919cd0e612ac1b811d9ae7e334c3ac Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Wed, 29 Aug 2018 18:47:25 +0800
Subject: [PATCH] fix comparison of index_t and size_t

---
 mace/kernels/unstack.h | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mace/kernels/unstack.h b/mace/kernels/unstack.h
index 08e3d798..82b5c467 100644
--- a/mace/kernels/unstack.h
+++ b/mace/kernels/unstack.h
@@ -40,14 +40,14 @@ struct UnstackFunctor {
     if (axis_ < 0) {
       axis_ += input->dim_size();
     }
-    MACE_CHECK(outputs.size() == input_shape[axis_],
+    MACE_CHECK(static_cast<index_t>(outputs.size()) == input_shape[axis_],
                "output size not equal input_shape[axis]");
 
     std::vector<index_t> output_shape = input_shape;
     output_shape.erase(output_shape.begin() + axis_);
 
     std::vector<T *> output_data(outputs.size(), nullptr);
-    for (size_t i = 0; i < input_shape[axis_]; ++i) {
+    for (index_t i = 0; i < input_shape[axis_]; ++i) {
       MACE_RETURN_IF_ERROR(outputs[i]->Resize(output_shape));
       output_data[i] = outputs[i]->mutable_data<T>();
     }
@@ -63,7 +63,7 @@ struct UnstackFunctor {
     for (index_t h = 0; h < high_dim_elem_size; ++h) {
       int input_idx = h * input_shape[axis_] * low_dim_elem_size;
       int output_idx = h * low_dim_elem_size;
-      for (size_t i = 0; i < input_shape[axis_]; ++i) {
+      for (index_t i = 0; i < input_shape[axis_]; ++i) {
         memcpy(output_data[i] + output_idx, input_data + input_idx,
                sizeof(T) * low_dim_elem_size);
         input_idx += low_dim_elem_size;
-- 
GitLab