From 75160d677975bf0063e3ef3b16c451e920781ed2 Mon Sep 17 00:00:00 2001 From: Fei Hu Date: Thu, 3 Mar 2022 01:24:44 -0800 Subject: [PATCH] Add the missing status return in beam search (#10738) --- .../contrib_ops/cuda/transformers/beam_search_device_helper.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/contrib_ops/cuda/transformers/beam_search_device_helper.cc b/onnxruntime/contrib_ops/cuda/transformers/beam_search_device_helper.cc index 068f6939ae..3b8497c32b 100644 --- a/onnxruntime/contrib_ops/cuda/transformers/beam_search_device_helper.cc +++ b/onnxruntime/contrib_ops/cuda/transformers/beam_search_device_helper.cc @@ -464,6 +464,7 @@ Status UpdateFeeds( // Make sure data is ready before next subgraph execution. CUDA_RETURN_IF_ERROR(cudaStreamSynchronize(reinterpret_cast(stream))); + return Status::OK(); } // Explicit template instantiations of functions