debug++

Tencent · Apr 23, 2024 · 98d77cf · 98d77cf
1 parent 5880edb
commit 98d77cf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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())