fix reduction fp16 exceed on android (#1819)

open-mmlab · Mar 1, 2023 · 84a289f · 84a289f
1 parent 29b29ee
commit 84a289f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/csrc/mmdeploy/backend_ops/ncnn/onnx2ncnn/onnx2ncnn.cpp b/csrc/mmdeploy/backend_ops/ncnn/onnx2ncnn/onnx2ncnn.cpp
@@ -2200,6 +2200,8 @@ int main(int argc, char** argv) {
       }
       fprintf(pp, " 4=%d", keepdims);
       fprintf(pp, " 5=1");
+      // Force set Reduction for FP32, FP16 may exceed for some models.
+      fprintf(pp, " 31=15");
     } else if (op == "Reorg") {
       int stride = get_node_attr_i(node, "stride", 1);
       fprintf(pp, " 0=%d", stride);