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 0) { + CUDA_RETURN_IF_ERROR(cudaStreamSynchronize(stream)); } // Assign TRT output back to ORT output