From 064197ca4340642cb6fb40572a1244b72a1cd435 Mon Sep 17 00:00:00 2001 From: lwkhahaha Date: Fri, 26 Jul 2024 15:13:30 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90Error=20Message=20No.=2077=20BUAA?= =?UTF-8?q?=E3=80=91rewrite=20error=20message?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/cpp/inference/api/analyzer_rnn1_tester.cc | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/test/cpp/inference/api/analyzer_rnn1_tester.cc b/test/cpp/inference/api/analyzer_rnn1_tester.cc index 72c53ccbdd8156..9a597fae58a1d6 100644 --- a/test/cpp/inference/api/analyzer_rnn1_tester.cc +++ b/test/cpp/inference/api/analyzer_rnn1_tester.cc @@ -55,8 +55,18 @@ struct DataRecord { CHECK(!data.link_step_data_all.empty()) << "empty"; CHECK(!data.week_data_all.empty()); CHECK(!data.minute_data_all.empty()); - CHECK_EQ(data.link_step_data_all.size(), data.week_data_all.size()); - CHECK_EQ(data.minute_data_all.size(), data.link_step_data_all.size()); + PADDLE_ENFORCE_EQ( + data.link_step_data_all.size(), + data.week_data_all.size(), + platform::errors::InvalidArgument( + "The value of data.link_step_data_all.size() is not equal to the " + "value of data.week_data_all.size().")) + PADDLE_ENFORCE_EQ( + data.minute_data_all.size(), + data.link_step_data_all.size(), + platform::errors::InvalidArgument( + "The value of data.minute_data_all.size() is not equal to the " + "value of data.link_step_data_all.size().")) for (size_t j = 0; j < data.link_step_data_all.size(); j++) { for (const auto &d : data.link_step_data_all[j]) { data.rnn_link_data.push_back(d);