mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix reduction fp16 exceed on android (#1819)

pull/1816/head
hanrui1sensetime 2023-03-01 13:40:20 +08:00 committed by GitHub
parent 29b29ee48a
commit 84a289fd88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/mmdeploy/backend_ops/ncnn/onnx2ncnn/onnx2ncnn.cpp
View File

@ -2200,6 +2200,8 @@ int main(int argc, char** argv) {
} }
fprintf(pp, " 4=%d", keepdims); fprintf(pp, " 4=%d", keepdims);
fprintf(pp, " 5=1"); fprintf(pp, " 5=1");
// Force set Reduction for FP32, FP16 may exceed for some models.
fprintf(pp, " 31=15");
} else if (op == "Reorg") { } else if (op == "Reorg") {
int stride = get_node_attr_i(node, "stride", 1); int stride = get_node_attr_i(node, "stride", 1);
fprintf(pp, " 0=%d", stride); fprintf(pp, " 0=%d", stride);