From 2622f627f84c41332960133b808416e51ebfe411 Mon Sep 17 00:00:00 2001 From: sihaixianyu Date: Tue, 27 Aug 2024 06:55:06 +0000 Subject: [PATCH] update src/core/key_process/key_process.cpp. [OPT] Add return status check for send eos signal failure. Signed-off-by: sihaixianyu --- src/core/key_process/key_process.cpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/core/key_process/key_process.cpp b/src/core/key_process/key_process.cpp index bc4b4496..a220aa44 100644 --- a/src/core/key_process/key_process.cpp +++ b/src/core/key_process/key_process.cpp @@ -1897,7 +1897,13 @@ void KeyProcess::SendEosTensor(const std::string& embName, int channel) size_t channelSize = 0; acltdtQueryChannelSize(transChannels[sendName], &channelSize); LOG_INFO("[EOS] Before send eos, channel:{}, size:{}.", sendName, channelSize); - SendTensorsByAcl(transChannels[sendName], ACL_TENSOR_DATA_END_OF_SEQUENCE, tensors, isNeedResend); + + auto status = SendTensorsByAcl(transChannels[sendName], ACL_TENSOR_DATA_END_OF_SEQUENCE, tensors, isNeedResend); + if (status != tensorflow::Status::OK()) { + LOG_ERROR("Failed to send eos signal to channel=>{}, got error: '{}'.", sendName, status.error_message()); + throw runtime_error("send eos signal failed!"); + } + acltdtQueryChannelSize(transChannels[sendName], &channelSize); LOG_INFO("[EOS] After send eos, channel:{}, size:{}.", sendName, channelSize); } -- Gitee