Fix sgemm bug

b446d200 · 李寅 · b1764247 · b446d200
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mace/kernels/sgemm.cc mace/kernels/sgemm.cc +1 -1

未找到文件。
--- a/mace/kernels/sgemm.cc
+++ b/mace/kernels/sgemm.cc
@@ -60,7 +60,7 @@ void SGemm::operator()(const MatrixMap<const float> &lhs,
          lhs.size() * sizeof(float)), DT_FLOAT));
    }
    if (!rhs.is_const()) {
-      packed_lhs_.reset(new Tensor(scratch_buffer->Scratch(
+      packed_rhs_.reset(new Tensor(scratch_buffer->Scratch(
          rhs.size() * sizeof(float)), DT_FLOAT));
    }
    packed_result_.reset(new Tensor(scratch_buffer->Scratch(