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

Add missing NEON ptranspose implementations.

Browse Source 
Unified implementation using only `vzip`.


(cherry picked from commit dba753a986)
This commit is contained in:
Antonio Sanchez
2021-05-24 21:34:35 -07:00
committed by Rasmus Munk Larsen
parent ee2a8f7139
commit 98cf1e076f
2 changed files with 241 additions and 328 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
test/packetmath.cpp
View File

@@ -546,22 +546,24 @@ void packetmath() {
}
}
const int m_size = PacketSize < 4 ? 1 : 4;
internal::PacketBlock<Packet, m_size> kernel2;
for (int i = 0; i < m_size; ++i) {
kernel2.packet[i] = internal::pload<Packet>(data1 + i * PacketSize);
}
ptranspose(kernel2);
int data_counter = 0;
for (int i = 0; i < PacketSize; ++i) {
for (int j = 0; j < m_size; ++j) {
data2[data_counter++] = data1[j*PacketSize + i];
// GeneralBlockPanelKernel also checks PacketBlock<Packet,(PacketSize%4)==0?4:PacketSize>;
if (PacketSize > 4 && PacketSize % 4 == 0) {
internal::PacketBlock<Packet, PacketSize%4==0?4:PacketSize> kernel2;
for (int i = 0; i < 4; ++i) {
kernel2.packet[i] = internal::pload<Packet>(data1 + i * PacketSize);
}
}
for (int i = 0; i < m_size; ++i) {
internal::pstore(data3, kernel2.packet[i]);
for (int j = 0; j < PacketSize; ++j) {
VERIFY(test::isApproxAbs(data3[j], data2[i*PacketSize + j], refvalue) && "ptranspose");
ptranspose(kernel2);
int data_counter = 0;
for (int i = 0; i < PacketSize; ++i) {
for (int j = 0; j < 4; ++j) {
data2[data_counter++] = data1[j*PacketSize + i];
}
}
for (int i = 0; i < 4; ++i) {
internal::pstore(data3, kernel2.packet[i]);
for (int j = 0; j < PacketSize; ++j) {
VERIFY(test::isApproxAbs(data3[j], data2[i*PacketSize + j], refvalue) && "ptranspose");
}
}
}