w

Tencent · Jan 23, 2025 · 7f5c391 · 7f5c391
1 parent f2a9d02
commit 7f5c391
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/layer/arm/convolution_arm.cpp b/src/layer/arm/convolution_arm.cpp
@@ -1376,7 +1376,14 @@ int Convolution_arm::forward_int8_arm(const Mat& bottom_blob, Mat& top_blob, con
 #if __ARM_NEON
     if (opt.use_packing_layout)
     {
-        out_elempack_int32 = num_output % 8 == 0 ? 8 : num_output % 4 == 0 ? 4 : 1;
+        if (ncnn::cpu_support_arm_asimdhp() && opt.use_fp16_arithmetic)
+        {
+            out_elempack_int32 = num_output % 8 == 0 ? 8 : num_output % 4 == 0 ? 4 : 1;
+        }
+        else
+        {
+            out_elempack_int32 = num_output % 4 == 0 ? 4 : 1;
+        }
     }
 #endif // __ARM_NEON