llvm-project/llvm/test/Transforms/InstCombine/bswap.ll

236 lines
6.6 KiB
LLVM

; RUN: opt < %s -instcombine -S | FileCheck %s
target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:128:128-n8:16:32"
define i32 @test1(i32 %i) {
; CHECK-LABEL: @test1(
; CHECK-NEXT: [[TMP12:%.*]] = call i32 @llvm.bswap.i32(i32 %i)
; CHECK-NEXT: ret i32 [[TMP12]]
;
%tmp1 = lshr i32 %i, 24
%tmp3 = lshr i32 %i, 8
%tmp4 = and i32 %tmp3, 65280
%tmp5 = or i32 %tmp1, %tmp4
%tmp7 = shl i32 %i, 8
%tmp8 = and i32 %tmp7, 16711680
%tmp9 = or i32 %tmp5, %tmp8
%tmp11 = shl i32 %i, 24
%tmp12 = or i32 %tmp9, %tmp11
ret i32 %tmp12
}
define i32 @test2(i32 %arg) {
; CHECK-LABEL: @test2(
; CHECK-NEXT: [[TMP14:%.*]] = call i32 @llvm.bswap.i32(i32 %arg)
; CHECK-NEXT: ret i32 [[TMP14]]
;
%tmp2 = shl i32 %arg, 24
%tmp4 = shl i32 %arg, 8
%tmp5 = and i32 %tmp4, 16711680
%tmp6 = or i32 %tmp2, %tmp5
%tmp8 = lshr i32 %arg, 8
%tmp9 = and i32 %tmp8, 65280
%tmp10 = or i32 %tmp6, %tmp9
%tmp12 = lshr i32 %arg, 24
%tmp14 = or i32 %tmp10, %tmp12
ret i32 %tmp14
}
define i16 @test3(i16 %s) {
; CHECK-LABEL: @test3(
; CHECK-NEXT: [[TMP5:%.*]] = call i16 @llvm.bswap.i16(i16 %s)
; CHECK-NEXT: ret i16 [[TMP5]]
;
%tmp2 = lshr i16 %s, 8
%tmp4 = shl i16 %s, 8
%tmp5 = or i16 %tmp2, %tmp4
ret i16 %tmp5
}
define i16 @test4(i16 %s) {
; CHECK-LABEL: @test4(
; CHECK-NEXT: [[TMP5:%.*]] = call i16 @llvm.bswap.i16(i16 %s)
; CHECK-NEXT: ret i16 [[TMP5]]
;
%tmp2 = lshr i16 %s, 8
%tmp4 = shl i16 %s, 8
%tmp5 = or i16 %tmp4, %tmp2
ret i16 %tmp5
}
define i16 @test5(i16 %a) {
; CHECK-LABEL: @test5(
; CHECK-NEXT: [[TMP_UPGRD_3:%.*]] = call i16 @llvm.bswap.i16(i16 %a)
; CHECK-NEXT: ret i16 [[TMP_UPGRD_3]]
;
%tmp = zext i16 %a to i32
%tmp1 = and i32 %tmp, 65280
%tmp2 = ashr i32 %tmp1, 8
%tmp2.upgrd.1 = trunc i32 %tmp2 to i16
%tmp4 = and i32 %tmp, 255
%tmp5 = shl i32 %tmp4, 8
%tmp5.upgrd.2 = trunc i32 %tmp5 to i16
%tmp.upgrd.3 = or i16 %tmp2.upgrd.1, %tmp5.upgrd.2
%tmp6 = bitcast i16 %tmp.upgrd.3 to i16
%tmp6.upgrd.4 = zext i16 %tmp6 to i32
%retval = trunc i32 %tmp6.upgrd.4 to i16
ret i16 %retval
}
; PR2842
define i32 @test6(i32 %x) nounwind readnone {
; CHECK-LABEL: @test6(
; CHECK-NEXT: [[TMP7:%.*]] = call i32 @llvm.bswap.i32(i32 %x)
; CHECK-NEXT: ret i32 [[TMP7]]
;
%tmp = shl i32 %x, 16
%x.mask = and i32 %x, 65280
%tmp1 = lshr i32 %x, 16
%tmp2 = and i32 %tmp1, 255
%tmp3 = or i32 %x.mask, %tmp
%tmp4 = or i32 %tmp3, %tmp2
%tmp5 = shl i32 %tmp4, 8
%tmp6 = lshr i32 %x, 24
%tmp7 = or i32 %tmp5, %tmp6
ret i32 %tmp7
}
declare void @extra_use(i32)
; swaphalf = (x << 16 | x >> 16)
; ((swaphalf & 0x00ff00ff) << 8) | ((swaphalf >> 8) & 0x00ff00ff)
define i32 @bswap32_and_first(i32 %x) {
; CHECK-LABEL: @bswap32_and_first(
; CHECK-NEXT: [[BSWAP:%.*]] = call i32 @llvm.bswap.i32(i32 %x)
; CHECK-NEXT: ret i32 [[BSWAP]]
;
%shl = shl i32 %x, 16
%shr = lshr i32 %x, 16
%swaphalf = or i32 %shl, %shr
%t = and i32 %swaphalf, 16711935
%tshl = shl nuw i32 %t, 8
%b = lshr i32 %swaphalf, 8
%band = and i32 %b, 16711935
%bswap = or i32 %tshl, %band
ret i32 %bswap
}
; FIXME: Extra use should not prevent matching to bswap.
; swaphalf = (x << 16 | x >> 16)
; ((swaphalf & 0x00ff00ff) << 8) | ((swaphalf >> 8) & 0x00ff00ff)
define i32 @bswap32_and_first_extra_use(i32 %x) {
; CHECK-LABEL: @bswap32_and_first_extra_use(
; CHECK-NEXT: [[SHL:%.*]] = shl i32 %x, 16
; CHECK-NEXT: [[SHR:%.*]] = lshr i32 %x, 16
; CHECK-NEXT: [[SWAPHALF:%.*]] = or i32 [[SHL]], [[SHR]]
; CHECK-NEXT: [[T:%.*]] = and i32 [[SWAPHALF]], 16711935
; CHECK-NEXT: [[TSHL:%.*]] = shl nuw i32 [[T]], 8
; CHECK-NEXT: [[B:%.*]] = lshr i32 [[SWAPHALF]], 8
; CHECK-NEXT: [[BAND:%.*]] = and i32 [[B]], 16711935
; CHECK-NEXT: [[BSWAP:%.*]] = or i32 [[TSHL]], [[BAND]]
; CHECK-NEXT: call void @extra_use(i32 [[T]])
; CHECK-NEXT: ret i32 [[BSWAP]]
;
%shl = shl i32 %x, 16
%shr = lshr i32 %x, 16
%swaphalf = or i32 %shl, %shr
%t = and i32 %swaphalf, 16711935
%tshl = shl nuw i32 %t, 8
%b = lshr i32 %swaphalf, 8
%band = and i32 %b, 16711935
%bswap = or i32 %tshl, %band
call void @extra_use(i32 %t)
ret i32 %bswap
}
; swaphalf = (x << 16 | x >> 16)
; ((swaphalf << 8) & 0xff00ff00) | ((swaphalf >> 8) & 0x00ff00ff)
; PR23863
define i32 @bswap32_shl_first(i32 %x) {
; CHECK-LABEL: @bswap32_shl_first(
; CHECK-NEXT: [[BSWAP:%.*]] = call i32 @llvm.bswap.i32(i32 %x)
; CHECK-NEXT: ret i32 [[BSWAP]]
;
%shl = shl i32 %x, 16
%shr = lshr i32 %x, 16
%swaphalf = or i32 %shl, %shr
%t = shl i32 %swaphalf, 8
%tand = and i32 %t, -16711936
%b = lshr i32 %swaphalf, 8
%band = and i32 %b, 16711935
%bswap = or i32 %tand, %band
ret i32 %bswap
}
; Extra use should not prevent matching to bswap.
; swaphalf = (x << 16 | x >> 16)
; ((swaphalf << 8) & 0xff00ff00) | ((swaphalf >> 8) & 0x00ff00ff)
define i32 @bswap32_shl_first_extra_use(i32 %x) {
; CHECK-LABEL: @bswap32_shl_first_extra_use(
; CHECK-NEXT: [[SHL:%.*]] = shl i32 %x, 16
; CHECK-NEXT: [[SHR:%.*]] = lshr i32 %x, 16
; CHECK-NEXT: [[SWAPHALF:%.*]] = or i32 [[SHL]], [[SHR]]
; CHECK-NEXT: [[T:%.*]] = shl i32 [[SWAPHALF]], 8
; CHECK-NEXT: [[BSWAP:%.*]] = call i32 @llvm.bswap.i32(i32 %x)
; CHECK-NEXT: call void @extra_use(i32 [[T]])
; CHECK-NEXT: ret i32 [[BSWAP]]
;
%shl = shl i32 %x, 16
%shr = lshr i32 %x, 16
%swaphalf = or i32 %shl, %shr
%t = shl i32 %swaphalf, 8
%tand = and i32 %t, -16711936
%b = lshr i32 %swaphalf, 8
%band = and i32 %b, 16711935
%bswap = or i32 %tand, %band
call void @extra_use(i32 %t)
ret i32 %bswap
}
define i16 @test8(i16 %a) {
; CHECK-LABEL: @test8(
; CHECK-NEXT: [[REV:%.*]] = call i16 @llvm.bswap.i16(i16 %a)
; CHECK-NEXT: ret i16 [[REV]]
;
%conv = zext i16 %a to i32
%shr = lshr i16 %a, 8
%shl = shl i32 %conv, 8
%conv1 = zext i16 %shr to i32
%or = or i32 %conv1, %shl
%conv2 = trunc i32 %or to i16
ret i16 %conv2
}
define i16 @test9(i16 %a) {
; CHECK-LABEL: @test9(
; CHECK-NEXT: [[REV:%.*]] = call i16 @llvm.bswap.i16(i16 %a)
; CHECK-NEXT: ret i16 [[REV]]
;
%conv = zext i16 %a to i32
%shr = lshr i32 %conv, 8
%shl = shl i32 %conv, 8
%or = or i32 %shr, %shl
%conv2 = trunc i32 %or to i16
ret i16 %conv2
}
define i16 @test10(i32 %a) {
; CHECK-LABEL: @test10(
; CHECK-NEXT: [[TRUNC:%.*]] = trunc i32 %a to i16
; CHECK-NEXT: [[REV:%.*]] = call i16 @llvm.bswap.i16(i16 [[TRUNC]])
; CHECK-NEXT: ret i16 [[REV]]
;
%shr1 = lshr i32 %a, 8
%and1 = and i32 %shr1, 255
%and2 = shl i32 %a, 8
%shl1 = and i32 %and2, 65280
%or = or i32 %and1, %shl1
%conv = trunc i32 %or to i16
ret i16 %conv
}