提交 b74ebd0b 编写于 作者: M mapingshuo 提交者: GitHub

fix sgemm bug, test=develop (#3053)

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