fix

Tencent · Jun 6, 2024 · dc16e08 · dc16e08
1 parent 04236ff
commit dc16e08
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/src/layer/arm/convolutiondepthwise_arm.cpp b/src/layer/arm/convolutiondepthwise_arm.cpp
@@ -1507,10 +1507,7 @@ int ConvolutionDepthWise_arm::forward_int8_arm(const Mat& bottom_blob, Mat& top_
 #if __ARM_NEON
     if (opt.use_packing_layout)
     {
-        if (use_int8_requantize)
-            out_elempack = num_output % 8 == 0 ? 8 : 1;
-        else
-            out_elempack = num_output % 4 == 0 ? 4 : 1;
+        out_elempack = num_output % 8 == 0 ? 8 : num_output % 4 == 0 ? 4 : 1;
     }
 #endif // __ARM_NEON
     size_t out_elemsize = use_int8_requantize ? 1u * out_elempack : 4u * out_elempack;
@@ -1537,10 +1534,7 @@ int ConvolutionDepthWise_arm::forward_int8_arm(const Mat& bottom_blob, Mat& top_
     if (opt.use_packing_layout)
     {
         g_elempack = channels_g % 8 == 0 ? 8 : 1;
-        if (use_int8_requantize)
-            out_g_elempack = num_output_g % 8 == 0 ? 8 : 1;
-        else
-            out_g_elempack = num_output_g % 4 == 0 ? 4 : 1;
+        out_g_elempack = num_output_g % 8 == 0 ? 8 : num_output_g % 4 == 0 ? 4 : 1;
     }
 #endif // __ARM_NEON