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

Improve logged message for nodes that are forced to execute on CPU rather than some other EP (usually CUDA) (#7297)

Browse source
This commit is contained in:
Hariharan Seshadri 2021-04-09 01:36:19 -07:00 committed by GitHub
parent 8ad5007f8f
commit 711cc99f4d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/framework/fallback_cpu_capability.cc
View file

@ -132,7 +132,9 @@ std::unordered_set<NodeIndex> GetCpuPreferredNodes(const onnxruntime::GraphViewe
if (place_in_cpu) {
cpu_nodes.insert(cur);
LOGS_DEFAULT(INFO) << "Force fallback to CPU execution for node: " << node->Name();
LOGS_DEFAULT(INFO) << "ORT optimization- Force fallback to CPU execution for node: " << node->Name()
<< " because the CPU execution path is deemed faster than overhead involved with execution on other EPs "
<< " capable of executing this node";
for (auto* output : node->OutputDefs()) {
cpu_output_args.insert(output);
}