mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-03-13 18:37:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Vectorized the packing of a row-major matrix used as the left hand side argument in a matrix-matrix product.

Browse Source
This commit is contained in:
Benoit Steiner 2014-03-27 10:02:24 -07:00
parent c4902a3d01
commit b776458ccb
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
Eigen/src/Core/products/GeneralBlockPanelKernel.h
View File

@ -954,9 +954,22 @@ EIGEN_DONT_INLINE void gemm_pack_lhs<Scalar, Index, Pack1, Pack2, StorageOrder,
}
else
{
for(Index k=0; k<depth; k++)
{
// TODO add a vectorized transpose here
const Index peeled_k = (depth/PacketSize)*PacketSize;
Index k=0;
for(; k<peeled_k; k+=PacketSize) {
for (Index m = 0; m < Pack1; m += PacketSize) {
Kernel<Packet> kernel;
for (int p = 0; p < PacketSize; ++p) {
kernel.packet[p] = ploadu<Packet>(&lhs(i+p+m, k));
}
ptranspose(kernel);
for (int p = 0; p < PacketSize; ++p) {
pstore(blockA+count+m+Pack1*p, cj.pconj(kernel.packet[p]));
}
}
count += PacketSize*Pack1;
}
for(; k<depth; k++) {
Index w=0;
for(; w<Pack1-3; w+=4)
{