From 0588d3f01304d44ca1ea28db44d7078be1395ead Mon Sep 17 00:00:00 2001
From: yf711 <yifanl@microsoft.com>
Date: Mon, 18 Dec 2023 12:54:00 -0800
Subject: [PATCH] fix

---
 .../core/providers/tensorrt/tensorrt_execution_provider.cc    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
index 0a5dd604001f5..95e4ace9fd268 100644
--- a/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
+++ b/onnxruntime/core/providers/tensorrt/tensorrt_execution_provider.cc
@@ -3395,8 +3395,8 @@ common::Status TensorrtExecutionProvider::Compile(const std::vector<FusedNodeAnd
         return ORT_MAKE_STATUS(ONNXRUNTIME, FAIL, "TensorRT EP execution context enqueue failed.");
       }
 
-      if (sync_stream_after_enqueue) {
-        cudaStreamSynchronize(stream);
+      if (sync_stream_after_enqueue_ || dds_output_set.size() > 0) {
+        CUDA_RETURN_IF_ERROR(cudaStreamSynchronize(stream));
       }
 
       // Assign TRT output back to ORT output