saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix a resolving issue on the quantization node transforming. (#7952)

Browse source
This commit is contained in:
Wenbing Li 2021-06-04 16:07:52 -07:00 committed by GitHub
parent b856e7ae3c
commit 09ab895563
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/optimizer/dynamic_quantize_matmul_fusion.cc
View file

@ -104,7 +104,7 @@ Status DynamicQuantizeMatMulFusion::ApplyImpl(Graph& graph, bool& modified, int
nodes_to_remove.push_back(matmul_integer_to_float_node);
}
modified = !nodes_to_remove.empty();
modified = modified || !nodes_to_remove.empty();
for (const auto& node : nodes_to_remove) {
graph_utils::RemoveNodeOutputEdges(graph, node);

2
onnxruntime/core/optimizer/matmul_integer_to_float.cc
View file

@ -153,7 +153,7 @@ Status MatMulIntegerToFloatFusion::ApplyImpl(Graph& graph, bool& modified, int g
}
}
modified = !nodes_to_remove.empty();
modified = modified || !nodes_to_remove.empty();
for (const auto& node : nodes_to_remove) {
graph_utils::RemoveNodeOutputEdges(graph, node);