From 4dae92c2a13a2956eb5d122833d0c79446fa28e5 Mon Sep 17 00:00:00 2001
From: Serge Panev <spanev@nvidia.com>
Date: Sat, 20 Jun 2020 13:56:33 -0700
Subject: [PATCH] Update the onnx-tensorrt submodule

Signed-off-by: Serge Panev <spanev@nvidia.com>
---
 3rdparty/onnx-tensorrt                             | 2 +-
 src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h  | 2 ++
 src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc | 4 ----
 3 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/3rdparty/onnx-tensorrt b/3rdparty/onnx-tensorrt
index f4745fcaff86..84b5be1d6fc0 160000
--- a/3rdparty/onnx-tensorrt
+++ b/3rdparty/onnx-tensorrt
@@ -1 +1 @@
-Subproject commit f4745fcaff868a519834917c657f105a8eef2f53
+Subproject commit 84b5be1d6fc03564f2c0dba85a2ee75bad242c2e
diff --git a/src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h b/src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h
index 2d133579f0d5..22ba18bccd06 100644
--- a/src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h
+++ b/src/operator/subgraph/tensorrt/nnvm_to_onnx-inl.h
@@ -31,6 +31,8 @@
 #include <mxnet/operator.h>
 #include <nnvm/pass_functions.h>
 
+
+#define ONNX_ML 1
 #include <onnx/onnx_pb.h>
 
 #include <unordered_map>
diff --git a/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc b/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc
index b02d1094183f..d82f7544a091 100644
--- a/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc
+++ b/src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc
@@ -35,13 +35,9 @@
 #include <google/protobuf/io/zero_copy_stream_impl.h>
 #include <google/protobuf/text_format.h>
 #include <onnx-tensorrt/NvOnnxParser.h>
-#include <onnx-tensorrt/NvOnnxParserRuntime.h>
 #include <dmlc/logging.h>
 #include <dmlc/parameter.h>
 
-#include <onnx-tensorrt/PluginFactory.hpp>
-#include <onnx-tensorrt/plugin_common.hpp>
-
 using std::cout;
 using std::cerr;
 using std::endl;