From a77f09c2a6fab73a991f2a16401172588b8b5260 Mon Sep 17 00:00:00 2001
From: zhaojiaying01 <zhaojiaying01@baidu.com>
Date: Mon, 6 Jan 2020 14:01:12 +0800
Subject: [PATCH] [Mobile][ARM]Fix bug to support PaddleSlim quantize trained
 model. test=mobile

---
 mobile/src/framework/executor.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/mobile/src/framework/executor.cpp b/mobile/src/framework/executor.cpp
index cda5c5522c..b64f5f4608 100644
--- a/mobile/src/framework/executor.cpp
+++ b/mobile/src/framework/executor.cpp
@@ -255,12 +255,14 @@ void Executor<Device, T>::LoadMemory(void **data,
     case VARTYPE_TYPE_INT8:
       LoadMemInternal<int8_t>(
           reinterpret_cast<void **>(data_buf),
-          reinterpret_cast<void *>(tensor->mutable_data<T>()), tensor->numel());
+          reinterpret_cast<void *>(tensor->mutable_data<int8_t>()),
+          tensor->numel());
       break;
     case VARTYPE_TYPE_INT32:
-      LoadMemInternal<int>(reinterpret_cast<void **>(data_buf),
-                           reinterpret_cast<void *>(tensor->mutable_data<T>()),
-                           tensor->numel());
+      LoadMemInternal<int>(
+          reinterpret_cast<void **>(data_buf),
+          reinterpret_cast<void *>(tensor->mutable_data<int>()),
+          tensor->numel());
       break;
     default:
       LOG(kLOG_ERROR) << "data type is not supported";
-- 
GitLab