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
 | |
| }
 |