sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Matrix mul bug

Browse Source
This commit is contained in:
kradchen
2023-05-16 11:01:01 +08:00
parent 3518ba36e9
commit 6106cc9a5a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Matrix.cpp
View File

@@ -884,7 +884,7 @@ namespace Aurora {
} else if (aOther.isVector()) { } else if (aOther.isVector()) {
// right size // right size
if (getDimSize(1) == aOther.getDimSize(0)) { if (getDimSize(1) == aOther.getDimSize(0)) {
auto result = deepCopy(); auto result = aOther.deepCopy();
cblas_dgemv(CblasColMajor, CblasNoTrans, getDimSize(0), cblas_dgemv(CblasColMajor, CblasNoTrans, getDimSize(0),
getDimSize(1), 1.0, getData(), getDimSize(1), 1.0, getData(),
getDimSize(0), aOther.getData(), 1, 0, getDimSize(0), aOther.getData(), 1, 0,