e6ad9ef4e7
The basic idea to this is that a) having a single canonical type makes CSE easier, and b) many of our transforms are inconsistent about which types we end up with based on visit order. I'm restricting this to constants as for non-constants, we'd have to decide whether the simplicity was worth extra instructions. For constants, there are no extra instructions. We chose the canonical type as i64 arbitrarily. We might consider changing this to something else in the future if we have cause. Differential Revision: https://reviews.llvm.org/D115387
196 lines
7.1 KiB
LLVM
196 lines
7.1 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
|
|
; RUN: opt < %s -instcombine -S -data-layout="e" | FileCheck %s --check-prefixes=ANY,LE
|
|
; RUN: opt < %s -instcombine -S -data-layout="E" | FileCheck %s --check-prefixes=ANY,BE
|
|
|
|
define i32 @shrinkExtractElt_i64_to_i32_0(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i32_0(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 0
|
|
; LE-NEXT: ret i32 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i32_0(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 1
|
|
; BE-NEXT: ret i32 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i32 0
|
|
%t = trunc i64 %e to i32
|
|
ret i32 %t
|
|
}
|
|
|
|
define i32 @vscale_shrinkExtractElt_i64_to_i32_0(<vscale x 3 x i64> %x) {
|
|
; LE-LABEL: @vscale_shrinkExtractElt_i64_to_i32_0(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <vscale x 3 x i64> [[X:%.*]] to <vscale x 6 x i32>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <vscale x 6 x i32> [[TMP1]], i64 0
|
|
; LE-NEXT: ret i32 [[T]]
|
|
;
|
|
; BE-LABEL: @vscale_shrinkExtractElt_i64_to_i32_0(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <vscale x 3 x i64> [[X:%.*]] to <vscale x 6 x i32>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <vscale x 6 x i32> [[TMP1]], i64 1
|
|
; BE-NEXT: ret i32 [[T]]
|
|
;
|
|
%e = extractelement <vscale x 3 x i64> %x, i32 0
|
|
%t = trunc i64 %e to i32
|
|
ret i32 %t
|
|
}
|
|
|
|
|
|
define i32 @shrinkExtractElt_i64_to_i32_1(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i32_1(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 2
|
|
; LE-NEXT: ret i32 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i32_1(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 3
|
|
; BE-NEXT: ret i32 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i32 1
|
|
%t = trunc i64 %e to i32
|
|
ret i32 %t
|
|
}
|
|
|
|
define i32 @shrinkExtractElt_i64_to_i32_2(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i32_2(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 4
|
|
; LE-NEXT: ret i32 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i32_2(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <6 x i32>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <6 x i32> [[TMP1]], i64 5
|
|
; BE-NEXT: ret i32 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i32 2
|
|
%t = trunc i64 %e to i32
|
|
ret i32 %t
|
|
}
|
|
|
|
define i16 @shrinkExtractElt_i64_to_i16_0(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i16_0(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 0
|
|
; LE-NEXT: ret i16 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i16_0(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 3
|
|
; BE-NEXT: ret i16 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i16 0
|
|
%t = trunc i64 %e to i16
|
|
ret i16 %t
|
|
}
|
|
|
|
define i16 @shrinkExtractElt_i64_to_i16_1(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i16_1(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 4
|
|
; LE-NEXT: ret i16 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i16_1(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 7
|
|
; BE-NEXT: ret i16 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i16 1
|
|
%t = trunc i64 %e to i16
|
|
ret i16 %t
|
|
}
|
|
|
|
define i16 @shrinkExtractElt_i64_to_i16_2(<3 x i64> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i64_to_i16_2(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 8
|
|
; LE-NEXT: ret i16 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i64_to_i16_2(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i64> [[X:%.*]] to <12 x i16>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <12 x i16> [[TMP1]], i64 11
|
|
; BE-NEXT: ret i16 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i16 2
|
|
%t = trunc i64 %e to i16
|
|
ret i16 %t
|
|
}
|
|
|
|
; Crazy types may be ok.
|
|
define i11 @shrinkExtractElt_i33_to_11_2(<3 x i33> %x) {
|
|
; LE-LABEL: @shrinkExtractElt_i33_to_11_2(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <3 x i33> [[X:%.*]] to <9 x i11>
|
|
; LE-NEXT: [[T:%.*]] = extractelement <9 x i11> [[TMP1]], i64 6
|
|
; LE-NEXT: ret i11 [[T]]
|
|
;
|
|
; BE-LABEL: @shrinkExtractElt_i33_to_11_2(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <3 x i33> [[X:%.*]] to <9 x i11>
|
|
; BE-NEXT: [[T:%.*]] = extractelement <9 x i11> [[TMP1]], i64 8
|
|
; BE-NEXT: ret i11 [[T]]
|
|
;
|
|
%e = extractelement <3 x i33> %x, i16 2
|
|
%t = trunc i33 %e to i11
|
|
ret i11 %t
|
|
}
|
|
|
|
; Do not optimize if it would result in an invalid bitcast instruction.
|
|
define i13 @shrinkExtractElt_i67_to_i13_2(<3 x i67> %x) {
|
|
; ANY-LABEL: @shrinkExtractElt_i67_to_i13_2(
|
|
; ANY-NEXT: [[E:%.*]] = extractelement <3 x i67> [[X:%.*]], i64 2
|
|
; ANY-NEXT: [[T:%.*]] = trunc i67 [[E]] to i13
|
|
; ANY-NEXT: ret i13 [[T]]
|
|
;
|
|
%e = extractelement <3 x i67> %x, i459 2
|
|
%t = trunc i67 %e to i13
|
|
ret i13 %t
|
|
}
|
|
|
|
; Do not optimize if the bitcast instruction would be valid, but the
|
|
; transform would be wrong.
|
|
define i30 @shrinkExtractElt_i40_to_i30_1(<3 x i40> %x) {
|
|
; ANY-LABEL: @shrinkExtractElt_i40_to_i30_1(
|
|
; ANY-NEXT: [[E:%.*]] = extractelement <3 x i40> [[X:%.*]], i64 1
|
|
; ANY-NEXT: [[T:%.*]] = trunc i40 [[E]] to i30
|
|
; ANY-NEXT: ret i30 [[T]]
|
|
;
|
|
%e = extractelement <3 x i40> %x, i32 1
|
|
%t = trunc i40 %e to i30
|
|
ret i30 %t
|
|
}
|
|
|
|
; Do not canonicalize if that would increase the instruction count.
|
|
declare void @use(i64)
|
|
define i16 @shrinkExtractElt_i64_to_i16_2_extra_use(<3 x i64> %x) {
|
|
; ANY-LABEL: @shrinkExtractElt_i64_to_i16_2_extra_use(
|
|
; ANY-NEXT: [[E:%.*]] = extractelement <3 x i64> [[X:%.*]], i64 2
|
|
; ANY-NEXT: call void @use(i64 [[E]])
|
|
; ANY-NEXT: [[T:%.*]] = trunc i64 [[E]] to i16
|
|
; ANY-NEXT: ret i16 [[T]]
|
|
;
|
|
%e = extractelement <3 x i64> %x, i64 2
|
|
call void @use(i64 %e)
|
|
%t = trunc i64 %e to i16
|
|
ret i16 %t
|
|
}
|
|
|
|
; Check to ensure PR45314 remains fixed.
|
|
define <4 x i64> @PR45314(<4 x i64> %x) {
|
|
; LE-LABEL: @PR45314(
|
|
; LE-NEXT: [[TMP1:%.*]] = bitcast <4 x i64> [[X:%.*]] to <8 x i32>
|
|
; LE-NEXT: [[S:%.*]] = shufflevector <8 x i32> [[TMP1]], <8 x i32> undef, <8 x i32> zeroinitializer
|
|
; LE-NEXT: [[B:%.*]] = bitcast <8 x i32> [[S]] to <4 x i64>
|
|
; LE-NEXT: ret <4 x i64> [[B]]
|
|
;
|
|
; BE-LABEL: @PR45314(
|
|
; BE-NEXT: [[TMP1:%.*]] = bitcast <4 x i64> [[X:%.*]] to <8 x i32>
|
|
; BE-NEXT: [[S:%.*]] = shufflevector <8 x i32> [[TMP1]], <8 x i32> undef, <8 x i32> <i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1, i32 1>
|
|
; BE-NEXT: [[B:%.*]] = bitcast <8 x i32> [[S]] to <4 x i64>
|
|
; BE-NEXT: ret <4 x i64> [[B]]
|
|
;
|
|
%e = extractelement <4 x i64> %x, i32 0
|
|
%t = trunc i64 %e to i32
|
|
%i = insertelement <8 x i32> undef, i32 %t, i32 0
|
|
%s = shufflevector <8 x i32> %i, <8 x i32> undef, <8 x i32> zeroinitializer
|
|
%b = bitcast <8 x i32> %s to <4 x i64>
|
|
ret <4 x i64> %b
|
|
}
|