From 2b49342a4d740554d5f5d8dbbe5fd14afa801a13 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Tue, 12 Dec 2023 15:02:34 +0800
Subject: [PATCH] Fix matrix MxA bug

---
 src/Matrix.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/Matrix.cpp b/src/Matrix.cpp
index 0fe5884..b480b58 100644
--- a/src/Matrix.cpp
+++ b/src/Matrix.cpp
@@ -54,7 +54,7 @@ namespace Aurora{
             aFunc(aMatrix.getDataSize(), aMatrix.getData()+1 , 2, &zero, 0, output+1 ,
                   2);
         } else{
-            aFunc(aMatrix.getDataSize(), aMatrix.getData(), 1, &aScalar, 0, output, 1);
+            aFunc(aMatrix.getDataSize()*aMatrix.getValueType(), aMatrix.getData(), 1, &aScalar, 0, output, 1);
         }
         return Matrix::New(output, aMatrix.getDimSize(0), aMatrix.getDimSize(1), aMatrix.getDimSize(2),
                                    aMatrix.getValueType());