未验证 提交 43a371fe 编写于 作者: M mapingshuo 提交者: GitHub

fix sgemm bug, test=develop (#3053)

上级 706a44d8
...@@ -126,7 +126,7 @@ void SequenceConvCompute::Run() { ...@@ -126,7 +126,7 @@ void SequenceConvCompute::Run() {
kernel_num, // ldb: n kernel_num, // ldb: n
0.f, // beta 0.f, // beta
out_data, // C out_data, // C
sequence_len, // ldc: m kernel_num, // ldc: n
NULL, // bias NULL, // bias
false, // is_bias false, // is_bias
act_param, // act_param act_param, // act_param
......
...@@ -106,7 +106,7 @@ class SequenceConvComputeTester : public arena::TestCase { ...@@ -106,7 +106,7 @@ class SequenceConvComputeTester : public arena::TestCase {
} }
for (int i = 0; i < output_shape[0]; i++) { for (int i = 0; i < output_shape[0]; i++) {
for (int j = 0; j < output_shape[1]; j++) { for (int j = 0; j < output_shape[1]; j++) {
output_data[i * output_shape[0] + j] = res[i][j]; output_data[i * output_shape[1] + j] = res[i][j];
} }
} }
(output->mutable_lod())->push_back(lod_[0]); (output->mutable_lod())->push_back(lod_[0]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册