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

[VITISAI] fix out of bound error on graph with loop (#17065)

Browse source 
### Description
<!-- Describe your changes. -->
Check the bound of the node_get_inputs for out of bound error.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
Model with loop would encounter this error. Currrent we do not support
custom op for loop. So, ideally it would throw an error and fall back to
CPU evalution.
This commit is contained in:
BoarQing 2023-08-10 09:38:30 +08:00 committed by GitHub
parent f17efb5c7b
commit e951f837e4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/core/providers/vitisai/imp/node.cc
View file

@ -21,7 +21,11 @@ vaip_core::DllSafe<std::vector<NodeInput>> node_get_inputs(const Node& node) {
}
for (auto iter = node.InputEdgesBegin(); iter != node.InputEdgesEnd();
++iter) {
ret[iter->GetDstArgIndex()].node = &iter->GetNode();
auto dst_idx = static_cast<size_t>(iter->GetDstArgIndex());
if (dst_idx < ret.size()) {
// ignore implicit nodes.
ret[dst_idx].node = &iter->GetNode();
}
}
return vaip_core::DllSafe(ret);
}