devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

WIP2

Browse Source
This commit is contained in:
Everton Constantino
2021-05-14 12:29:37 +00:00
parent 9fc17867e5
commit ad67705447
1 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
Eigen/src/Core/arch/NEON/Kernels.h
View File

@@ -423,20 +423,50 @@ struct Accumulator<0, CPU, Scalar, ResScalar, DataMapper, 12, 4>
LinearMapper r2 = dest.getLinearMapper(row, col + 2); LinearMapper r2 = dest.getLinearMapper(row, col + 2);
LinearMapper r3 = dest.getLinearMapper(row, col + 3); LinearMapper r3 = dest.getLinearMapper(row, col + 3);
r0.storePacket(0*PacketSize, r0.template loadPacket<ResPacket>(0*PacketSize) + pAlpha*_acc1.packet[0]); ResPacket R00 = r0.template loadPacket<ResPacket>(0*PacketSize);
r1.storePacket(0*PacketSize, r1.template loadPacket<ResPacket>(0*PacketSize) + pAlpha*_acc1.packet[1]); ResPacket R01 = r1.template loadPacket<ResPacket>(0*PacketSize);
r2.storePacket(0*PacketSize, r2.template loadPacket<ResPacket>(0*PacketSize) + pAlpha*_acc1.packet[2]); ResPacket R02 = r2.template loadPacket<ResPacket>(0*PacketSize);
r3.storePacket(0*PacketSize, r3.template loadPacket<ResPacket>(0*PacketSize) + pAlpha*_acc1.packet[3]); ResPacket R03 = r3.template loadPacket<ResPacket>(0*PacketSize);
r0.storePacket(1*PacketSize, r0.template loadPacket<ResPacket>(1*PacketSize) + pAlpha*_acc2.packet[0]); ResPacket R10 = r0.template loadPacket<ResPacket>(1*PacketSize);
r1.storePacket(1*PacketSize, r1.template loadPacket<ResPacket>(1*PacketSize) + pAlpha*_acc2.packet[1]); ResPacket R11 = r1.template loadPacket<ResPacket>(1*PacketSize);
r2.storePacket(1*PacketSize, r2.template loadPacket<ResPacket>(1*PacketSize) + pAlpha*_acc2.packet[2]); ResPacket R12 = r2.template loadPacket<ResPacket>(1*PacketSize);
r3.storePacket(1*PacketSize, r3.template loadPacket<ResPacket>(1*PacketSize) + pAlpha*_acc2.packet[3]); ResPacket R13 = r3.template loadPacket<ResPacket>(1*PacketSize);
r0.storePacket(2*PacketSize, r0.template loadPacket<ResPacket>(2*PacketSize) + pAlpha*_acc3.packet[0]); ResPacket R20 = r0.template loadPacket<ResPacket>(2*PacketSize);
r1.storePacket(2*PacketSize, r1.template loadPacket<ResPacket>(2*PacketSize) + pAlpha*_acc3.packet[1]); ResPacket R21 = r1.template loadPacket<ResPacket>(2*PacketSize);
r2.storePacket(2*PacketSize, r2.template loadPacket<ResPacket>(2*PacketSize) + pAlpha*_acc3.packet[2]); ResPacket R22 = r2.template loadPacket<ResPacket>(2*PacketSize);
r3.storePacket(2*PacketSize, r3.template loadPacket<ResPacket>(2*PacketSize) + pAlpha*_acc3.packet[3]); ResPacket R23 = r3.template loadPacket<ResPacket>(2*PacketSize);
R00 += pAlpha*_acc1.packet[0];
R01 += pAlpha*_acc1.packet[1];
R02 += pAlpha*_acc1.packet[2];
R03 += pAlpha*_acc1.packet[3];
R10 += pAlpha*_acc2.packet[0];
R11 += pAlpha*_acc2.packet[1];
R12 += pAlpha*_acc2.packet[2];
R13 += pAlpha*_acc2.packet[3];
R20 += pAlpha*_acc3.packet[0];
R21 += pAlpha*_acc3.packet[1];
R22 += pAlpha*_acc3.packet[2];
R23 += pAlpha*_acc3.packet[3];
r0.storePacket(0*PacketSize, R00);
r1.storePacket(0*PacketSize, R01);
r2.storePacket(0*PacketSize, R02);
r3.storePacket(0*PacketSize, R03);
r0.storePacket(1*PacketSize, R10);
r1.storePacket(1*PacketSize, R11);
r2.storePacket(1*PacketSize, R12);
r3.storePacket(1*PacketSize, R13);
r0.storePacket(2*PacketSize, R20);
r1.storePacket(2*PacketSize, R21);
r2.storePacket(2*PacketSize, R22);
r3.storePacket(2*PacketSize, R23);
} }
}; };