From 10c04bce22c28b9adad1b72696e396773a359718 Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Wed, 16 Jan 2019 06:10:29 -0800 Subject: [PATCH 1/3] [MXNET-703] Update onnx-tensorrt for fp16 support --- 3rdparty/onnx-tensorrt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/3rdparty/onnx-tensorrt b/3rdparty/onnx-tensorrt index f1c7aa63d88d..1e209e546061 160000 --- a/3rdparty/onnx-tensorrt +++ b/3rdparty/onnx-tensorrt @@ -1 +1 @@ -Subproject commit f1c7aa63d88d8d8ef70490f2ebb6b33f7450218b +Subproject commit 1e209e546061173ccc37b25bbca69a795c6c86e4 From 8f990b0060313f8c3958ae2d8ff56819f760b426 Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Thu, 17 Jan 2019 08:42:25 -0800 Subject: [PATCH 2/3] Add Cuda headers to include path The Docker image cuda:10.0-cudnn7-devel has moved around a few paths and as a result some cuda headers are no longer in a default C++ header search path. This makes sure they're availible when building. --- ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt | 1 + 1 file changed, 1 insertion(+) diff --git a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt index f4844115c0fd..8ad90aedeb6f 100644 --- a/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt +++ b/ci/docker/Dockerfile.build.ubuntu_gpu_tensorrt @@ -39,3 +39,4 @@ COPY runtime_functions.sh /work/ WORKDIR /work/mxnet ENV LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/lib +ENV CPLUS_INCLUDE_PATH=${CPLUS_INCLUDE_PATH}:/usr/local/cuda-10.0/targets/x86_64-linux/include/ From 2d55cf24819d8019ba2a51e0e6d8b8d9c1da42e5 Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Sat, 26 Jan 2019 12:43:12 -0800 Subject: [PATCH 3/3] Incorporate a few more bug fixes --- 3rdparty/onnx-tensorrt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/3rdparty/onnx-tensorrt b/3rdparty/onnx-tensorrt index 1e209e546061..7823c0cca3ef 160000 --- a/3rdparty/onnx-tensorrt +++ b/3rdparty/onnx-tensorrt @@ -1 +1 @@ -Subproject commit 1e209e546061173ccc37b25bbca69a795c6c86e4 +Subproject commit 7823c0cca3efd27c26bbcaea479d306432d6f53d