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

[TVM] Bugfix. (#117)

Browse source
This commit is contained in:
Siva 2018-12-08 00:01:16 +05:30 committed by Raymond Yang
parent 23e122f304
commit 19c630b87f
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/test/tvm/tvm_basic_test.cc
View file

@ -82,7 +82,7 @@ void FuseAdd(const onnxruntime::GraphViewer& graph, std::vector<std::unique_ptr<
for (int i = 0; i < add_nodes.size(); ++i) {
auto node = graph.GetNode(add_nodes[i]);
for (auto it = node->InputNodesBegin(); it != node->InputNodesEnd(); ++it) {
auto index_it = std::find(add_nodes.begin(), add_nodes.end(), (*it)->Index());
auto index_it = std::find(add_nodes.begin(), add_nodes.end(), (*it).Index());
if (index_it != add_nodes.end()) {
set.merge(i, static_cast<int>(index_it - add_nodes.begin()));
}