Address comments

nod-ai · Sep 20, 2024 · de47766 · de47766
1 parent 8c17229
commit de47766
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 17 deletions.
diff --git a/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.cpp b/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.cpp
@@ -228,23 +228,6 @@ Status GraphInfo::Initialize() {
 }
 
 const onnx::TypeProto *GraphInfo::FindTypeProtoForName(std::string_view name) {
-  // Node outputs don't typically have type information, but shape inference
-  // will associate them in the value_info. If not there, it may be a
-  // graph output, which must have type information.
-#if 0
-  {
-    auto it = value_info_map_.find(name);
-    if (it != value_info_map_.end()) {
-      return &it->second.type();
-    }
-  }
-  {
-    auto it = output_map_.find(name);
-    if (it != output_map_.end()) {
-      return &it->second.type();
-    }
-  }
-#endif
   return graph_viewer_.GetNodeArg(std::string{name})->TypeAsProto();
 }
 

diff --git a/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.h b/onnxruntime/core/providers/iree/compiler/torch-mlir-import-onnx/OnnxImporter.h
@@ -22,6 +22,7 @@
 #include "onnx/onnx_pb.h"
 
 #include <optional>
+#include <memory>
 #include <string_view>
 #include <unordered_map>