saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-19 21:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor fixes (#12276)

Browse source 
* Log message for each node placement so the output won't get truncated.

* Use correct variable in ReshapeOpBuilder::CanSkipReshape().
This commit is contained in:
Edward Chen 2022-07-22 14:33:49 -07:00 committed by GitHub
parent 0264a9c29b
commit cb351388d0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
onnxruntime/core/framework/session_state.cc
View file

@ -195,7 +195,7 @@ Status SessionState::CreateKernels(const KernelRegistryManager& kernel_registry_
return Status::OK();
}
const SequentialExecutionPlan* SessionState::GetExecutionPlan() const {
const SequentialExecutionPlan* SessionState::GetExecutionPlan() const {
if (!p_seq_exec_plan_.has_value()) {
return nullptr;
}
@ -1185,13 +1185,14 @@ static Status VerifyEachNodeIsAssignedToAnEp(const Graph& graph, const logging::
if (is_verbose_mode) {
LOGS(logger, VERBOSE) << "Node placements";
if (node_placements.size() == 1) {
LOGS(logger, VERBOSE) << "All nodes have been placed on [" << node_placements.begin()->first << "].";
const auto& [provider, node_strs] = *node_placements.begin();
LOGS(logger, VERBOSE) << " All nodes placed on [" << provider << "]. Number of nodes: " << node_strs.size();
} else {
for (const auto& [provider, node_strs] : node_placements) {
std::ostringstream all_nodes_str;
std::copy(node_strs.begin(), node_strs.end(), std::ostream_iterator<std::string>(all_nodes_str, ", "));
LOGS(logger, VERBOSE) << " Provider: [" << provider << "]"
<< ": [" << all_nodes_str.str() << "]";
LOGS(logger, VERBOSE) << " Node(s) placed on [" << provider << "]. Number of nodes: " << node_strs.size();
for (const auto& node_str : node_strs) {
LOGS(logger, VERBOSE) << " " << node_str;
}
}
}
}

2
onnxruntime/core/providers/nnapi/nnapi_builtin/builders/op_builder.cc
View file

@ -827,7 +827,7 @@ bool ReshapeOpBuilder::IsQuantizedOp(const NodeUnit& node_unit) const {
}
// Now the dest node is Gemm/Matmul, we want to make sure it is supported
if (!BaseOpBuilder::IsOpSupported(model_builder, node_unit)) {
if (!BaseOpBuilder::IsOpSupported(model_builder, dest_node_unit)) {
return false;
}