admin/omath
1
0
Fork 0
mirror of https://github.com/orange-cpp/omath.git synced 2026-02-13 07:03:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #71 from orange-cpp/feature/mat_perf_boost

Browse Source 
Improves matrix multiplication performance
This commit is contained in:
Orange++
2025-09-17 17:18:12 +03:00
committed by GitHub
parent a3a023a664 e01d32fb22
commit 04b50d4545
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
include/omath/linear_algebra/mat.hpp
View File

@@ -157,14 +157,25 @@ namespace omath
{
Mat<Rows, OtherColumns, Type, StoreType> result;
for (size_t i = 0; i < Rows; ++i)
for (size_t j = 0; j < OtherColumns; ++j)
{
Type sum = 0;
if constexpr (StoreType == MatStoreType::ROW_MAJOR)
for (size_t i = 0; i < Rows; ++i)
for (size_t k = 0; k < Columns; ++k)
sum += at(i, k) * other.at(k, j);
result.at(i, j) = sum;
}
{
const Type aik = at(i, k);
for (size_t j = 0; j < OtherColumns; ++j)
result.at(i, j) += aik * other.at(k, j);
}
else if constexpr (StoreType == MatStoreType::COLUMN_MAJOR)
for (size_t j = 0; j < OtherColumns; ++j)
for (size_t k = 0; k < Columns; ++k)
{
const Type bkj = other.at(k, j);
for (size_t i = 0; i < Rows; ++i)
result.at(i, j) += at(i, k) * bkj;
}
else
std::unreachable();
return result;
}