diff --git a/mace/kernels/sgemm.cc b/mace/kernels/sgemm.cc index e1be6f99cb1ea81072b07c85b208ef53153ab1c5..ab8310d1eb4221ee4b42ed64a15c944bcb0add8a 100644 --- a/mace/kernels/sgemm.cc +++ b/mace/kernels/sgemm.cc @@ -60,7 +60,7 @@ void SGemm::operator()(const MatrixMap &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(