From dfb0f010f1b19d7ab6c5eb657d2fd7cbf16e6ab4 Mon Sep 17 00:00:00 2001
From: liutuo <liutuo@xiaomi.com>
Date: Wed, 13 Jun 2018 09:46:48 +0800
Subject: [PATCH] fix reshape test bug

---
 mace/ops/reshape_test.cc | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/mace/ops/reshape_test.cc b/mace/ops/reshape_test.cc
index 4e48d384..91c0f82b 100644
--- a/mace/ops/reshape_test.cc
+++ b/mace/ops/reshape_test.cc
@@ -36,9 +36,10 @@ void TestReshape(const std::vector<index_t> &org_shape,
 
   // Add input data
   net.AddRandomInput<DeviceType::CPU, float>("Input", org_shape);
-  net.AddInputFromArray<DeviceType::CPU, int32_t>("Shape",
-                                                  {output_shape.size()},
-                                                  output_shape);
+  net.AddInputFromArray<DeviceType::CPU, int32_t>(
+      "Shape",
+      {static_cast<index_t>(output_shape.size())},
+      output_shape);
 
   // Run
   net.RunOp();
-- 
GitLab