From ef6b812e3874838d91f03b2025b32983baf393e6 Mon Sep 17 00:00:00 2001 From: Serge Panev Date: Tue, 23 Jun 2020 18:48:25 -0700 Subject: [PATCH] Update the onnx-tensorrt submodule Signed-off-by: Serge Panev --- 3rdparty/onnx-tensorrt | 2 +- Makefile | 4 ++-- src/operator/subgraph/tensorrt/onnx_to_tensorrt.cc | 4 ---- 3 files changed, 3 insertions(+), 7 deletions(-) diff --git a/3rdparty/onnx-tensorrt b/3rdparty/onnx-tensorrt index f4745fcaff86..ff2e75415806 160000 --- a/3rdparty/onnx-tensorrt +++ b/3rdparty/onnx-tensorrt @@ -1 +1 @@ -Subproject commit f4745fcaff868a519834917c657f105a8eef2f53 +Subproject commit ff2e75415806afcdd809b71fe65d3bfd12452f3c diff --git a/Makefile b/Makefile index 51930c782391..ae88d68ed9a2 100644 --- a/Makefile +++ b/Makefile @@ -129,8 +129,8 @@ ifeq ($(USE_NVTX), 1) endif ifeq ($(USE_TENSORRT), 1) - CFLAGS += -I$(ROOTDIR) -I$(TPARTYDIR) -DONNX_NAMESPACE=$(ONNX_NAMESPACE) -DMXNET_USE_TENSORRT=1 - LDFLAGS += -lprotobuf -pthread -lonnx -lonnx_proto -lnvonnxparser -lnvonnxparser_runtime -lnvinfer -lnvinfer_plugin + CFLAGS += -I$(ROOTDIR) -I$(TPARTYDIR) -DONNX_NAMESPACE=$(ONNX_NAMESPACE) -DMXNET_USE_TENSORRT=1 -DONNX_ML=1 + LDFLAGS += -lprotobuf -pthread -lonnx -lonnx_proto -lnvonnxparser -lnvinfer -lnvinfer_plugin endif # -L/usr/local/lib 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 #include #include -#include #include #include -#include -#include - using std::cout; using std::cerr; using std::endl;