Skip to content

Commit 68cb903

Browse files
committed
Revert d43ec97 "[X86] combineConcatVectorOps - IsConcatFree - peek through bitcasts to find inplace subvectors."
I've been given reports of this causing infinite loops downstream - I'm going to revert for now while I investigate.
1 parent b634e05 commit 68cb903

File tree

2 files changed

+22
-12
lines changed

2 files changed

+22
-12
lines changed

llvm/lib/Target/X86/X86ISelLowering.cpp

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -56163,19 +56163,18 @@ static SDValue combineConcatVectorOps(const SDLoc &DL, MVT VT,
5616356163
};
5616456164
auto IsConcatFree = [](MVT VT, ArrayRef<SDValue> SubOps, unsigned Op) {
5616556165
bool AllConstants = true;
56166-
bool AllSubs = true;
56167-
unsigned VecSize = VT.getSizeInBits();
56166+
bool AllSubVectors = true;
5616856167
for (unsigned I = 0, E = SubOps.size(); I != E; ++I) {
56169-
SDValue BC = peekThroughBitcasts(SubOps[I].getOperand(Op));
56170-
unsigned SubSize = BC.getValueSizeInBits();
56171-
unsigned EltSize = BC.getScalarValueSizeInBits();
56168+
SDValue Sub = SubOps[I].getOperand(Op);
56169+
unsigned NumSubElts = Sub.getValueType().getVectorNumElements();
56170+
SDValue BC = peekThroughBitcasts(Sub);
5617256171
AllConstants &= ISD::isBuildVectorOfConstantSDNodes(BC.getNode()) ||
5617356172
ISD::isBuildVectorOfConstantFPSDNodes(BC.getNode());
56174-
AllSubs &= BC.getOpcode() == ISD::EXTRACT_SUBVECTOR &&
56175-
BC.getOperand(0).getValueSizeInBits() == VecSize &&
56176-
(BC.getConstantOperandVal(1) * EltSize) == (I * SubSize);
56173+
AllSubVectors &= Sub.getOpcode() == ISD::EXTRACT_SUBVECTOR &&
56174+
Sub.getOperand(0).getValueType() == VT &&
56175+
Sub.getConstantOperandAPInt(1) == (I * NumSubElts);
5617756176
}
56178-
return AllConstants || AllSubs;
56177+
return AllConstants || AllSubVectors;
5617956178
};
5618056179

5618156180
switch (Op0.getOpcode()) {

llvm/test/CodeGen/X86/vselect-avx.ll

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,7 @@ define void @blendv_split(ptr %p, <8 x i32> %cond, <8 x i32> %a, <8 x i32> %x, <
259259
ret void
260260
}
261261

262-
; Concatenate 128-bit pblendvb back together on AVX2+ targets (hidden by SSE __m128i bitcasts)
262+
; TODO: Concatenate 128-bit pblendvb back together on AVX2+ targets (hidden by SSE __m128i bitcasts)
263263
define <4 x i64> @vselect_concat_split_v16i8(<4 x i64> %a, <4 x i64> %b, <4 x i64> %c, <4 x i64> %d) {
264264
; AVX1-LABEL: vselect_concat_split_v16i8:
265265
; AVX1: ## %bb.0:
@@ -277,13 +277,24 @@ define <4 x i64> @vselect_concat_split_v16i8(<4 x i64> %a, <4 x i64> %b, <4 x i6
277277
; AVX2-LABEL: vselect_concat_split_v16i8:
278278
; AVX2: ## %bb.0:
279279
; AVX2-NEXT: vpcmpgtb %ymm2, %ymm3, %ymm2
280-
; AVX2-NEXT: vpblendvb %ymm2, %ymm1, %ymm0, %ymm0
280+
; AVX2-NEXT: vpblendvb %xmm2, %xmm1, %xmm0, %xmm3
281+
; AVX2-NEXT: vextracti128 $1, %ymm0, %xmm0
282+
; AVX2-NEXT: vextracti128 $1, %ymm1, %xmm1
283+
; AVX2-NEXT: vextracti128 $1, %ymm2, %xmm2
284+
; AVX2-NEXT: vpblendvb %xmm2, %xmm1, %xmm0, %xmm0
285+
; AVX2-NEXT: vinserti128 $1, %xmm0, %ymm3, %ymm0
281286
; AVX2-NEXT: retq
282287
;
283288
; AVX512-LABEL: vselect_concat_split_v16i8:
284289
; AVX512: ## %bb.0:
285290
; AVX512-NEXT: vpcmpgtb %ymm2, %ymm3, %ymm2
286-
; AVX512-NEXT: vpternlogq $216, %ymm2, %ymm1, %ymm0
291+
; AVX512-NEXT: vextracti128 $1, %ymm2, %xmm3
292+
; AVX512-NEXT: vextracti128 $1, %ymm1, %xmm4
293+
; AVX512-NEXT: ## kill: def $xmm1 killed $xmm1 killed $ymm1 def $ymm1
294+
; AVX512-NEXT: vpternlogq $226, %xmm0, %xmm2, %xmm1
295+
; AVX512-NEXT: vextracti128 $1, %ymm0, %xmm0
296+
; AVX512-NEXT: vpternlogq $226, %xmm0, %xmm3, %xmm4
297+
; AVX512-NEXT: vinserti128 $1, %xmm4, %ymm1, %ymm0
287298
; AVX512-NEXT: retq
288299
%a.bc = bitcast <4 x i64> %a to <32 x i8>
289300
%b.bc = bitcast <4 x i64> %b to <32 x i8>

0 commit comments

Comments
 (0)