From 98d77cfd5bef4c15be5c9d3fe005e61f36a806af Mon Sep 17 00:00:00 2001 From: nihuini Date: Tue, 23 Apr 2024 19:41:37 +0800 Subject: [PATCH] debug++ --- src/layer/arm/rnn_arm.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/layer/arm/rnn_arm.cpp b/src/layer/arm/rnn_arm.cpp index 64294a0c274..895f16bb913 100644 --- a/src/layer/arm/rnn_arm.cpp +++ b/src/layer/arm/rnn_arm.cpp @@ -332,6 +332,8 @@ static int rnn_int8(const Mat& bottom_blob, Mat& top_blob, int reverse, const Ma int num_output = top_blob.w; + NCNN_LOGE("rnn_int8 %d %d %d", size, T, num_output); + // num_output Mat gates(num_output, 4u, opt.workspace_allocator); if (gates.empty()) @@ -1049,6 +1051,8 @@ static int rnn_bf16s_int8(const Mat& bottom_blob, Mat& top_blob, int reverse, co int num_output = top_blob.w; + NCNN_LOGE("rnn_bf16s_int8 %d %d %d", size, T, num_output); + // num_output Mat gates(num_output, 4u, opt.workspace_allocator); if (gates.empty())