!4324 fix int8 conv non-optimize bug

Merge pull request !4324 from fuzhiye/tmp
5 years ago · 13c2b23356
parent 869f885122 3cda158a84
commit 13c2b23356
1 changed files with 3 additions and 1 deletions
--- a/mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/nnacl/pack.cc
@ -74,7 +74,9 @@ void PackWeightInt8(int8_t *weight_data, ConvParameter *conv_param, int8_t *pack

  for (int m = 0; m < kernel_plane; m++) {
    int kernel_plane_stride = m * in_channel;
-    int packed_kernel_plane_stride = m * C4NUM;
+    int plane_block = m / C4NUM;
+    int plane_res = m % C4NUM;
+    int packed_kernel_plane_stride = plane_block * C4NUM * C4NUM * ic4 * C4NUM + plane_res * C4NUM;
    for (int i = 0; i < ic4; i++) {
      int channel_block_stride = kernel_plane_stride + i * C4NUM;
      int packed_channel_block_size = packed_kernel_plane_stride + i * C4NUM * C4NUM * C4NUM;