From bcbcb2552affdbb4fda6f3daa3590edb97fbd022 Mon Sep 17 00:00:00 2001
From: stevenlix <38092805+stevenlix@users.noreply.github.com>
Date: Fri, 9 Oct 2020 21:49:40 -0700
Subject: [PATCH] update onnx-tensorrt submodule (#5442)

---
 cmake/external/onnx-tensorrt                                    | 2 +-
 .../core/providers/tensorrt/tensorrt_execution_provider.cc      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/cmake/external/onnx-tensorrt b/cmake/external/onnx-tensorrt
index fbebb14474..a3a4e38b2d 160000
--- a/cmake/external/onnx-tensorrt
+++ b/cmake/external/onnx-tensorrt
@@ -1 +1 @@
-Subproject commit fbebb144744b3be8defecd8478d74940056df305
+Subproject commit a3a4e38b2dfa7a62b6dcae33c0d1678b3bb5ef2a
diff --git a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
index 1e9d8a9918..964b37a72c 100644
--- a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
+++ b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
@@ -1337,6 +1337,7 @@ common::Status TensorrtExecutionProvider::Provider_Compile(const std::vector<onn
         }
         return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, "TensorRT EP execution context enqueue failed.");
       }
+      cudaDeviceSynchronize();
 
       // Cast INT64 input to INT32 because TensorRT doesn't fully support INT64
       for (int i = 0, end = output_binding_names.size(); i < end; ++i) {
@@ -1355,7 +1356,6 @@ common::Status TensorrtExecutionProvider::Provider_Compile(const std::vector<onn
         }
       }
 
-      cudaDeviceSynchronize();
       for (const auto& binding_index : binding_buffers_to_freeup) {
         cudaFree(buffers[binding_index]);
       }