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

Added op_name to message when we are missing a kernel. (#8110)

Browse source 
* Added op_name to message when we are missing a kernel.

* Added domain and version

* Added missing ,
This commit is contained in:
Chandru Ramakrishnan 2021-06-21 14:45:53 -04:00 committed by GitHub
parent cba4bc11c7
commit 10b7ed6430
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/eager/ort_kernel_invoker.cc
View file

@ -16,7 +16,7 @@ common::Status ORTInvoker::Invoke(const std::string& op_name,
std::vector<OrtValue>& outputs,
const NodeAttributes* attributes,
const std::string& domain,
const int /*version*/) {
const int version) {
//create a graph
Model model("test",
false,
@ -61,7 +61,7 @@ common::Status ORTInvoker::Invoke(const std::string& op_name,
OptimizerExecutionFrame::Info info({&node}, initializer_map, graph.ModelPath(), *execution_provider_);
auto kernel = info.CreateKernel(&node);
if (!kernel) {
ORT_THROW("Could not find kernel");
ORT_THROW("Could not find kernel name:", op_name, ", domain:", domain, ", version:", version);
}
std::vector<int> fetch_mlvalue_idxs;