GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix number of block columns to NOT overflow the cache (PowerPC) abnormally in GEMV

Browse Source
This commit is contained in:
Chip Kerchner 2022-01-27 20:35:53 +00:00 committed by Rasmus Munk Larsen
parent 7db0ac977a
commit 66464bd2a8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Eigen/src/Core/arch/AltiVec/MatrixVectorProduct.h
View File

@ -416,7 +416,7 @@ EIGEN_STRONG_INLINE void gemv_col(
#endif #endif
// TODO: improve the following heuristic: // TODO: improve the following heuristic:
const Index block_cols = cols < 128 ? cols : (lhsStride * sizeof(LhsScalar) < 32000 ? 16 : 16); const Index block_cols = cols < 128 ? cols : (lhsStride * sizeof(LhsScalar) < 16000 ? 16 : 8);
ResPacket palpha = pset1<ResPacket>(alpha); ResPacket palpha = pset1<ResPacket>(alpha);
for (Index j2 = 0; j2 < cols; j2 += block_cols) for (Index j2 = 0; j2 < cols; j2 += block_cols)
@ -1622,7 +1622,7 @@ EIGEN_STRONG_INLINE void gemv_complex_col(
const Index n1 = rows - 1 * ResPacketSize + 1; const Index n1 = rows - 1 * ResPacketSize + 1;
// TODO: improve the following heuristic: // TODO: improve the following heuristic:
const Index block_cols = cols < 128 ? cols : (lhsStride * sizeof(LhsScalar) < 32000 ? 16 : 16); const Index block_cols = cols < 128 ? cols : (lhsStride * sizeof(LhsScalar) < 16000 ? 16 : 8);
typedef alpha_store<PResPacket, ResPacket, ResScalar, Scalar> AlphaData; typedef alpha_store<PResPacket, ResPacket, ResScalar, Scalar> AlphaData;
AlphaData alpha_data(alpha); AlphaData alpha_data(alpha);