From 689c947f86d1398ba487816530accb8bd8af52b7 Mon Sep 17 00:00:00 2001 From: guohongzilong <2713219276@qq.com> Date: Mon, 9 Nov 2020 21:08:41 +0800 Subject: [PATCH] fix dequant memory invalid write --- mindspore/lite/src/runtime/kernel/arm/base/dequant.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/base/dequant.h b/mindspore/lite/src/runtime/kernel/arm/base/dequant.h index 7edf3c133f..1363fb0515 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/dequant.h +++ b/mindspore/lite/src/runtime/kernel/arm/base/dequant.h @@ -122,9 +122,9 @@ class DequantUtil { uint_result = 0; (*count)++; } - if (is_last) { - int remainder = unpack_bit_data->size(); - for (int i = 0; i < remainder; i++) { + size_t remainder = unpack_bit_data->size(); + if (is_last && remainder > 0) { + for (size_t i = 0; i < remainder; i++) { bool bit = unpack_bit_data->front(); uint_result = (static_cast(bit) << i) + uint_result; unpack_bit_data->pop();