From b446d20016a718e2de68268db1455a177f9738c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=85?= <liyin@xiaomi.com>
Date: Wed, 12 Sep 2018 16:58:21 +0800
Subject: [PATCH] Fix sgemm bug

---
 mace/kernels/sgemm.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/kernels/sgemm.cc b/mace/kernels/sgemm.cc
index e1be6f99..ab8310d1 100644
--- 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(
-- 
GitLab