[llvm] r311260 - [ARM] Factorize the calculation of WhichResult in isV*Mask. NFC.
Martin Storsjo via llvm-commits
llvm-commits at lists.llvm.org
Sat Aug 19 13:26:51 PDT 2017
Author: mstorsjo
Date: Sat Aug 19 13:26:51 2017
New Revision: 311260
URL: http://llvm.org/viewvc/llvm-project?rev=311260&view=rev
Log:
[ARM] Factorize the calculation of WhichResult in isV*Mask. NFC.
Differential Revision: https://reviews.llvm.org/D36930
Modified:
llvm/trunk/lib/Target/ARM/ARMISelLowering.cpp
Modified: llvm/trunk/lib/Target/ARM/ARMISelLowering.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/ARM/ARMISelLowering.cpp?rev=311260&r1=311259&r2=311260&view=diff
==============================================================================
--- llvm/trunk/lib/Target/ARM/ARMISelLowering.cpp (original)
+++ llvm/trunk/lib/Target/ARM/ARMISelLowering.cpp Sat Aug 19 13:26:51 2017
@@ -5800,6 +5800,13 @@ static bool isVTBLMask(ArrayRef<int> M,
return VT == MVT::v8i8 && M.size() == 8;
}
+static unsigned SelectPairHalf(unsigned Elements, ArrayRef<int> Mask,
+ unsigned Index) {
+ if (Mask.size() == Elements * 2)
+ return Index / Elements;
+ return Mask[Index] == 0 ? 0 : 1;
+}
+
// Checks whether the shuffle mask represents a vector transpose (VTRN) by
// checking that pairs of elements in the shuffle mask represent the same index
// in each vector, incrementing the expected index by 2 at each step.
@@ -5836,10 +5843,7 @@ static bool isVTRNMask(ArrayRef<int> M,
// element is undefined, e.g. [-1, 4, 2, 6] will be rejected, because only
// M[0] is used to determine WhichResult
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
for (unsigned j = 0; j < NumElts; j += 2) {
if ((M[i+j] >= 0 && (unsigned) M[i+j] != j + WhichResult) ||
(M[i+j+1] >= 0 && (unsigned) M[i+j+1] != j + NumElts + WhichResult))
@@ -5866,10 +5870,7 @@ static bool isVTRN_v_undef_Mask(ArrayRef
return false;
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
for (unsigned j = 0; j < NumElts; j += 2) {
if ((M[i+j] >= 0 && (unsigned) M[i+j] != j + WhichResult) ||
(M[i+j+1] >= 0 && (unsigned) M[i+j+1] != j + WhichResult))
@@ -5901,10 +5902,7 @@ static bool isVUZPMask(ArrayRef<int> M,
return false;
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
for (unsigned j = 0; j < NumElts; ++j) {
if (M[i+j] >= 0 && (unsigned) M[i+j] != 2 * j + WhichResult)
return false;
@@ -5935,10 +5933,7 @@ static bool isVUZP_v_undef_Mask(ArrayRef
unsigned Half = NumElts / 2;
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
for (unsigned j = 0; j < NumElts; j += Half) {
unsigned Idx = WhichResult;
for (unsigned k = 0; k < Half; ++k) {
@@ -5978,10 +5973,7 @@ static bool isVZIPMask(ArrayRef<int> M,
return false;
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
unsigned Idx = WhichResult * NumElts / 2;
for (unsigned j = 0; j < NumElts; j += 2) {
if ((M[i+j] >= 0 && (unsigned) M[i+j] != Idx) ||
@@ -6014,10 +6006,7 @@ static bool isVZIP_v_undef_Mask(ArrayRef
return false;
for (unsigned i = 0; i < M.size(); i += NumElts) {
- if (M.size() == NumElts * 2)
- WhichResult = i / NumElts;
- else
- WhichResult = M[i] == 0 ? 0 : 1;
+ WhichResult = SelectPairHalf(NumElts, M, i);
unsigned Idx = WhichResult * NumElts / 2;
for (unsigned j = 0; j < NumElts; j += 2) {
if ((M[i+j] >= 0 && (unsigned) M[i+j] != Idx) ||
More information about the llvm-commits
mailing list