From 19c630b87f1f74940f2b50c03bbb4ab7030a58ba Mon Sep 17 00:00:00 2001 From: Siva Date: Sat, 8 Dec 2018 00:01:16 +0530 Subject: [PATCH] [TVM] Bugfix. (#117) --- onnxruntime/test/tvm/tvm_basic_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/test/tvm/tvm_basic_test.cc b/onnxruntime/test/tvm/tvm_basic_test.cc index c266535b2d..5144280089 100644 --- a/onnxruntime/test/tvm/tvm_basic_test.cc +++ b/onnxruntime/test/tvm/tvm_basic_test.cc @@ -82,7 +82,7 @@ void FuseAdd(const onnxruntime::GraphViewer& graph, std::vectorInputNodesBegin(); 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(index_it - add_nodes.begin())); }