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

Improves matrix multiplication performance

Browse Source 
Optimizes matrix multiplication by specializing the algorithm
based on the matrix storage type (row-major or column-major).

This change significantly improves performance by leveraging
memory access patterns specific to each storage order.
This commit is contained in:
Orange
2025-09-17 17:12:41 +03:00
parent a3a023a664
commit e01d32fb22
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;
}