llvm/test/Transforms/InstCombine/trunc.ll

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s -passes=instcombine -S | FileCheck %s
target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64"

; Instcombine should be able to eliminate all of these ext casts.

declare void @use(i32)
declare void @use.i8(i8)
declare void @use_vec(<2 x i32>)

define i64 @test1(i64 %a) {
; CHECK-LABEL: @test1(
; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32
; CHECK-NEXT:    [[C:%.*]] = and i64 [[A]], 15
; CHECK-NEXT:    call void @use(i32 [[B]])
; CHECK-NEXT:    ret i64 [[C]]
;
  %b = trunc i64 %a to i32
  %c = and i32 %b, 15
  %d = zext i32 %c to i64
  call void @use(i32 %b)
  ret i64 %d
}

define <2 x i64> @test1_vec(<2 x i64> %a) {
; CHECK-LABEL: @test1_vec(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D:%.*]] = and <2 x i64> [[A]], splat (i64 15)
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = and <2 x i32> %b, <i32 15, i32 15>
  %d = zext <2 x i32> %c to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define <2 x i64> @test1_vec_nonuniform(<2 x i64> %a) {
; CHECK-LABEL: @test1_vec_nonuniform(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D:%.*]] = and <2 x i64> [[A]], <i64 15, i64 7>
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = and <2 x i32> %b, <i32 15, i32 7>
  %d = zext <2 x i32> %c to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define <2 x i64> @test1_vec_poison(<2 x i64> %a) {
; CHECK-LABEL: @test1_vec_poison(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D:%.*]] = and <2 x i64> [[A]], <i64 15, i64 poison>
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = and <2 x i32> %b, <i32 15, i32 poison>
  %d = zext <2 x i32> %c to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define i64 @test2(i64 %a) {
; CHECK-LABEL: @test2(
; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32
; CHECK-NEXT:    [[D1:%.*]] = shl i64 [[A]], 36
; CHECK-NEXT:    [[D:%.*]] = ashr exact i64 [[D1]], 36
; CHECK-NEXT:    call void @use(i32 [[B]])
; CHECK-NEXT:    ret i64 [[D]]
;
  %b = trunc i64 %a to i32
  %c = shl i32 %b, 4
  %q = ashr i32 %c, 4
  %d = sext i32 %q to i64
  call void @use(i32 %b)
  ret i64 %d
}

define <2 x i64> @test2_vec(<2 x i64> %a) {
; CHECK-LABEL: @test2_vec(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D1:%.*]] = shl <2 x i64> [[A]], splat (i64 36)
; CHECK-NEXT:    [[D:%.*]] = ashr exact <2 x i64> [[D1]], splat (i64 36)
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = shl <2 x i32> %b, <i32 4, i32 4>
  %q = ashr <2 x i32> %c, <i32 4, i32 4>
  %d = sext <2 x i32> %q to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define <2 x i64> @test2_vec_nonuniform(<2 x i64> %a) {
; CHECK-LABEL: @test2_vec_nonuniform(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D1:%.*]] = shl <2 x i64> [[A]], <i64 36, i64 37>
; CHECK-NEXT:    [[D:%.*]] = ashr exact <2 x i64> [[D1]], <i64 36, i64 37>
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = shl <2 x i32> %b, <i32 4, i32 5>
  %q = ashr <2 x i32> %c, <i32 4, i32 5>
  %d = sext <2 x i32> %q to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define <2 x i64> @test2_vec_poison(<2 x i64> %a) {
; CHECK-LABEL: @test2_vec_poison(
; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
; CHECK-NEXT:    [[D1:%.*]] = shl <2 x i64> [[A]], <i64 36, i64 poison>
; CHECK-NEXT:    [[D:%.*]] = ashr exact <2 x i64> [[D1]], <i64 36, i64 poison>
; CHECK-NEXT:    call void @use_vec(<2 x i32> [[B]])
; CHECK-NEXT:    ret <2 x i64> [[D]]
;
  %b = trunc <2 x i64> %a to <2 x i32>
  %c = shl <2 x i32> %b, <i32 4, i32 poison>
  %q = ashr <2 x i32> %c, <i32 4, i32 poison>
  %d = sext <2 x i32> %q to <2 x i64>
  call void @use_vec(<2 x i32> %b)
  ret <2 x i64> %d
}

define i64 @test3(i64 %a) {
; CHECK-LABEL: @test3(
; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32
; CHECK-NEXT:    [[C:%.*]] = and i64 [[A]], 8
; CHECK-NEXT:    call void @use(i32 [[B]])
; CHECK-NEXT:    ret i64 [[C]]
;
  %b = trunc i64 %a to i32
  %c = and i32 %b, 8
  %d = zext i32 %c to i64
  call void @use(i32 %b)
  ret i64 %d
}

define i64 @test4(i64 %a) {
; CHECK-LABEL: @test4(
; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32
; CHECK-NEXT:    [[C:%.*]] = and i64 [[A]], 8
; CHECK-NEXT:    [[X:%.*]] = xor i64 [[C]], 8
; CHECK-NEXT:    call void @use(i32 [[B]])
; CHECK-NEXT:    ret i64 [[X]]
;
  %b = trunc i64 %a to i32
  %c = and i32 %b, 8
  %x = xor i32 %c, 8
  %d = zext i32 %x to i64
  call void @use(i32 %b)
  ret i64 %d
}

define i32 @test5(i32 %A) {
; CHECK-LABEL: @test5(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr i32 [[A:%.*]], 16
; CHECK-NEXT:    ret i32 [[TMP1]]
;
  %B = zext i32 %A to i128
  %C = lshr i128 %B, 16
  %D = trunc i128 %C to i32
  ret i32 %D
}

define i32 @test6(i64 %A) {
; CHECK-LABEL: @test6(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr i64 [[A:%.*]], 32
; CHECK-NEXT:    [[D:%.*]] = trunc nuw i64 [[TMP1]] to i32
; CHECK-NEXT:    ret i32 [[D]]
;
  %B = zext i64 %A to i128
  %C = lshr i128 %B, 32
  %D = trunc i128 %C to i32
  ret i32 %D
}

; Test case where 'ashr' demanded bits does not contain any of the high bits,
; but does contain sign bits, where the sign bit is not known to be zero.
define i16 @ashr_mul_sign_bits(i8 %X, i8 %Y) {
; CHECK-LABEL: @ashr_mul_sign_bits(
; CHECK-NEXT:    [[A:%.*]] = sext i8 [[X:%.*]] to i16
; CHECK-NEXT:    [[B:%.*]] = sext i8 [[Y:%.*]] to i16
; CHECK-NEXT:    [[C:%.*]] = mul nsw i16 [[A]], [[B]]
; CHECK-NEXT:    [[D:%.*]] = ashr i16 [[C]], 3
; CHECK-NEXT:    ret i16 [[D]]
;
  %A = sext i8 %X to i32
  %B = sext i8 %Y to i32
  %C = mul i32 %A, %B
  %D = ashr i32 %C, 3
  %E = trunc i32 %D to i16
  ret i16 %E
}

define i16 @ashr_mul(i8 %X, i8 %Y) {
; CHECK-LABEL: @ashr_mul(
; CHECK-NEXT:    [[A:%.*]] = sext i8 [[X:%.*]] to i16
; CHECK-NEXT:    [[B:%.*]] = sext i8 [[Y:%.*]] to i16
; CHECK-NEXT:    [[C:%.*]] = mul nsw i16 [[A]], [[B]]
; CHECK-NEXT:    [[D:%.*]] = ashr i16 [[C]], 8
; CHECK-NEXT:    ret i16 [[D]]
;
  %A = sext i8 %X to i20
  %B = sext i8 %Y to i20
  %C = mul i20 %A, %B
  %D = ashr i20 %C, 8
  %E = trunc i20 %D to i16
  ret i16 %E
}

define i32 @trunc_ashr(i32 %X) {
; CHECK-LABEL: @trunc_ashr(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr i32 [[X:%.*]], 8
; CHECK-NEXT:    [[TMP2:%.*]] = or i32 [[TMP1]], -8388608
; CHECK-NEXT:    ret i32 [[TMP2]]
;
  %A = zext i32 %X to i36
  %B = or i36 %A, -2147483648 ; 0xF80000000
  %C = ashr i36 %B, 8
  %T = trunc i36 %C to i32
  ret i32  %T
}

define <2 x i32> @trunc_ashr_vec(<2 x i32> %X) {
; CHECK-LABEL: @trunc_ashr_vec(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr <2 x i32> [[X:%.*]], splat (i32 8)
; CHECK-NEXT:    [[TMP2:%.*]] = or <2 x i32> [[TMP1]], splat (i32 -8388608)
; CHECK-NEXT:    ret <2 x i32> [[TMP2]]
;
  %A = zext <2 x i32> %X to <2 x i36>
  %B = or <2 x i36> %A, <i36 -2147483648, i36 -2147483648> ; 0xF80000000
  %C = ashr <2 x i36> %B, <i36 8, i36 8>
  %T = trunc <2 x i36> %C to <2 x i32>
  ret <2 x i32>  %T
}

define i92 @test7(i64 %A) {
; CHECK-LABEL: @test7(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr i64 [[A:%.*]], 32
; CHECK-NEXT:    [[D:%.*]] = zext nneg i64 [[TMP1]] to i92
; CHECK-NEXT:    ret i92 [[D]]
;
  %B = zext i64 %A to i128
  %C = lshr i128 %B, 32
  %D = trunc i128 %C to i92
  ret i92 %D
}

define i64 @test8(i32 %A, i32 %B) {
; CHECK-LABEL: @test8(
; CHECK-NEXT:    [[C:%.*]] = zext i32 [[A:%.*]] to i64
; CHECK-NEXT:    [[D:%.*]] = zext i32 [[B:%.*]] to i64
; CHECK-NEXT:    [[E:%.*]] = shl nuw i64 [[D]], 32
; CHECK-NEXT:    [[F:%.*]] = or disjoint i64 [[E]], [[C]]
; CHECK-NEXT:    ret i64 [[F]]
;
  %C = zext i32 %A to i128
  %D = zext i32 %B to i128
  %E = shl i128 %D, 32
  %F = or i128 %E, %C
  %G = trunc i128 %F to i64
  ret i64 %G
}

define <2 x i64> @test8_vec(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test8_vec(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[D:%.*]] = zext <2 x i32> [[B:%.*]] to <2 x i64>
; CHECK-NEXT:    [[E:%.*]] = shl nuw <2 x i64> [[D]], splat (i64 32)
; CHECK-NEXT:    [[F:%.*]] = or disjoint <2 x i64> [[E]], [[C]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = shl <2 x i128> %D, <i128 32, i128 32>
  %F = or <2 x i128> %E, %C
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test8_vec_nonuniform(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test8_vec_nonuniform(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[D:%.*]] = zext <2 x i32> [[B:%.*]] to <2 x i64>
; CHECK-NEXT:    [[E:%.*]] = shl <2 x i64> [[D]], <i64 32, i64 48>
; CHECK-NEXT:    [[F:%.*]] = or disjoint <2 x i64> [[E]], [[C]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = shl <2 x i128> %D, <i128 32, i128 48>
  %F = or <2 x i128> %E, %C
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test8_vec_poison(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test8_vec_poison(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[D:%.*]] = zext <2 x i32> [[B:%.*]] to <2 x i64>
; CHECK-NEXT:    [[E:%.*]] = shl nuw <2 x i64> [[D]], <i64 32, i64 poison>
; CHECK-NEXT:    [[F:%.*]] = or disjoint <2 x i64> [[E]], [[C]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = shl <2 x i128> %D, <i128 32, i128 poison>
  %F = or <2 x i128> %E, %C
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define i8 @test9(i32 %X) {
; CHECK-LABEL: @test9(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[X:%.*]] to i8
; CHECK-NEXT:    [[Z:%.*]] = and i8 [[TMP1]], 42
; CHECK-NEXT:    ret i8 [[Z]]
;
  %Y = and i32 %X, 42
  %Z = trunc i32 %Y to i8
  ret i8 %Z
}

; rdar://8808586
define i8 @test10(i32 %X) {
; CHECK-LABEL: @test10(
; CHECK-NEXT:    [[Y:%.*]] = trunc i32 [[X:%.*]] to i8
; CHECK-NEXT:    [[Z:%.*]] = and i8 [[Y]], 42
; CHECK-NEXT:    ret i8 [[Z]]
;
  %Y = trunc i32 %X to i8
  %Z = and i8 %Y, 42
  ret i8 %Z
}

define i64 @test11(i32 %A, i32 %B) {
; CHECK-LABEL: @test11(
; CHECK-NEXT:    [[C:%.*]] = zext i32 [[A:%.*]] to i64
; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[B:%.*]], 31
; CHECK-NEXT:    [[E:%.*]] = zext nneg i32 [[TMP1]] to i64
; CHECK-NEXT:    [[F:%.*]] = shl nuw nsw i64 [[C]], [[E]]
; CHECK-NEXT:    ret i64 [[F]]
;
  %C = zext i32 %A to i128
  %D = zext i32 %B to i128
  %E = and i128 %D, 31
  %F = shl i128 %C, %E
  %G = trunc i128 %F to i64
  ret i64 %G
}

define <2 x i64> @test11_vec(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test11_vec(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], splat (i32 31)
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = shl nuw nsw <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 31>
  %F = shl <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test11_vec_nonuniform(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test11_vec_nonuniform(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 15>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = shl nuw nsw <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 15>
  %F = shl <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test11_vec_poison(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test11_vec_poison(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 poison>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = shl nuw nsw <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 poison>
  %F = shl <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define i64 @test12(i32 %A, i32 %B) {
; CHECK-LABEL: @test12(
; CHECK-NEXT:    [[C:%.*]] = zext i32 [[A:%.*]] to i64
; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[B:%.*]], 31
; CHECK-NEXT:    [[E:%.*]] = zext nneg i32 [[TMP1]] to i64
; CHECK-NEXT:    [[F:%.*]] = lshr i64 [[C]], [[E]]
; CHECK-NEXT:    ret i64 [[F]]
;
  %C = zext i32 %A to i128
  %D = zext i32 %B to i128
  %E = and i128 %D, 31
  %F = lshr i128 %C, %E
  %G = trunc i128 %F to i64
  ret i64 %G
}

define <2 x i64> @test12_vec(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test12_vec(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], splat (i32 31)
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = lshr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 31>
  %F = lshr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test12_vec_nonuniform(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test12_vec_nonuniform(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 15>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = lshr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 15>
  %F = lshr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test12_vec_poison(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test12_vec_poison(
; CHECK-NEXT:    [[C:%.*]] = zext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 poison>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = lshr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = zext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 poison>
  %F = lshr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define i64 @test13(i32 %A, i32 %B) {
; CHECK-LABEL: @test13(
; CHECK-NEXT:    [[C:%.*]] = sext i32 [[A:%.*]] to i64
; CHECK-NEXT:    [[TMP1:%.*]] = and i32 [[B:%.*]], 31
; CHECK-NEXT:    [[E:%.*]] = zext nneg i32 [[TMP1]] to i64
; CHECK-NEXT:    [[F:%.*]] = ashr i64 [[C]], [[E]]
; CHECK-NEXT:    ret i64 [[F]]
;
  %C = sext i32 %A to i128
  %D = zext i32 %B to i128
  %E = and i128 %D, 31
  %F = ashr i128 %C, %E
  %G = trunc i128 %F to i64
  ret i64 %G
}

define <2 x i64> @test13_vec(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test13_vec(
; CHECK-NEXT:    [[C:%.*]] = sext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], splat (i32 31)
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = ashr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = sext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 31>
  %F = ashr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test13_vec_nonuniform(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test13_vec_nonuniform(
; CHECK-NEXT:    [[C:%.*]] = sext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 15>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = ashr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = sext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 15>
  %F = ashr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

define <2 x i64> @test13_vec_poison(<2 x i32> %A, <2 x i32> %B) {
; CHECK-LABEL: @test13_vec_poison(
; CHECK-NEXT:    [[C:%.*]] = sext <2 x i32> [[A:%.*]] to <2 x i64>
; CHECK-NEXT:    [[TMP1:%.*]] = and <2 x i32> [[B:%.*]], <i32 31, i32 poison>
; CHECK-NEXT:    [[E:%.*]] = zext nneg <2 x i32> [[TMP1]] to <2 x i64>
; CHECK-NEXT:    [[F:%.*]] = ashr <2 x i64> [[C]], [[E]]
; CHECK-NEXT:    ret <2 x i64> [[F]]
;
  %C = sext <2 x i32> %A to <2 x i128>
  %D = zext <2 x i32> %B to <2 x i128>
  %E = and <2 x i128> %D, <i128 31, i128 poison>
  %F = ashr <2 x i128> %C, %E
  %G = trunc <2 x i128> %F to <2 x i64>
  ret <2 x i64> %G
}

; PR25543
; https://llvm.org/bugs/show_bug.cgi?id=25543
; This is an extractelement.

define i32 @trunc_bitcast1(<4 x i32> %v) {
; CHECK-LABEL: @trunc_bitcast1(
; CHECK-NEXT:    [[EXT:%.*]] = extractelement <4 x i32> [[V:%.*]], i64 1
; CHECK-NEXT:    ret i32 [[EXT]]
;
  %bc = bitcast <4 x i32> %v to i128
  %shr = lshr i128 %bc, 32
  %ext = trunc i128 %shr to i32
  ret i32 %ext
}

; A bitcast may still be required.

define i32 @trunc_bitcast2(<2 x i64> %v) {
; CHECK-LABEL: @trunc_bitcast2(
; CHECK-NEXT:    [[BC1:%.*]] = bitcast <2 x i64> [[V:%.*]] to <4 x i32>
; CHECK-NEXT:    [[EXT:%.*]] = extractelement <4 x i32> [[BC1]], i64 2
; CHECK-NEXT:    ret i32 [[EXT]]
;
  %bc = bitcast <2 x i64> %v to i128
  %shr = lshr i128 %bc, 64
  %ext = trunc i128 %shr to i32
  ret i32 %ext
}

; The right shift is optional.

define i32 @trunc_bitcast3(<4 x i32> %v) {
; CHECK-LABEL: @trunc_bitcast3(
; CHECK-NEXT:    [[EXT:%.*]] = extractelement <4 x i32> [[V:%.*]], i64 0
; CHECK-NEXT:    ret i32 [[EXT]]
;
  %bc = bitcast <4 x i32> %v to i128
  %ext = trunc i128 %bc to i32
  ret i32 %ext
}

define i32 @trunc_shl_31_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_31_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 31
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl i64 %val, 31
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_nsw_31_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_nsw_31_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 31
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nsw i64 %val, 31
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_nuw_31_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_nuw_31_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 31
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nuw i64 %val, 31
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_nsw_nuw_31_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_nsw_nuw_31_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 31
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nsw nuw i64 %val, 31
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_nuw_shl_nuw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_shl_nuw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc nuw i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl nuw i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nuw i64 %val, 3
  %trunc = trunc nuw i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_nsw_shl_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nsw_shl_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc nsw i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl nsw i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nsw i64 %val, 3
  %trunc = trunc nsw i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_nuw_nsw_shl_nuw_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_nsw_shl_nuw_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc nuw nsw i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl nuw nsw i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nuw nsw i64 %val, 3
  %trunc = trunc nuw nsw i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_nuw_shl_nuw_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_shl_nuw_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc nuw i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl nuw i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nuw nsw i64 %val, 3
  %trunc = trunc nuw i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_nuw_nsw_shl_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_nsw_shl_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc nsw i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl nsw i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nsw i64 %val, 3
  %trunc = trunc nuw nsw i64 %shl to i32
  ret i32 %trunc
}

; Missing flags on shl.
define i32 @trunc_nuw_shl_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_shl_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl i64 %val, 3
  %trunc = trunc nuw i64 %shl to i32
  ret i32 %trunc
}

; Missing flags on shl.
define i32 @trunc_shl_nuw_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_nuw_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nuw nsw i64 %val, 3
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

; Mismatched flags on shl/trunc.
define i32 @trunc_nuw_shl_nsw_3_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_nuw_shl_nsw_3_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 3
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl nsw i64 %val, 3
  %trunc = trunc nuw i64 %shl to i32
  ret i32 %trunc
}

define i4 @trunc_nuw_shl_i6_to_i4(i6 %x, i6 %y) {
; CHECK-LABEL: @trunc_nuw_shl_i6_to_i4(
; CHECK-NEXT:    [[X_TR:%.*]] = trunc i6 [[X:%.*]] to i4
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i4 [[X_TR]], 3
; CHECK-NEXT:    ret i4 [[TRUNC]]
;
  %shl = shl i6 %x, 3
  %trunc = trunc nuw i6 %shl to i4
  ret i4 %trunc
}

define i16 @trunc_shl_15_i16_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_15_i16_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i16
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i16 [[VAL_TR]], 15
; CHECK-NEXT:    ret i16 [[TRUNC]]
;
  %shl = shl i64 %val, 15
  %trunc = trunc i64 %shl to i16
  ret i16 %trunc
}

define i16 @trunc_shl_15_i16_i32(i32 %val) {
; CHECK-LABEL: @trunc_shl_15_i16_i32(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i32 [[VAL:%.*]] to i16
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i16 [[VAL_TR]], 15
; CHECK-NEXT:    ret i16 [[TRUNC]]
;
  %shl = shl i32 %val, 15
  %trunc = trunc i32 %shl to i16
  ret i16 %trunc
}

define i8 @trunc_shl_7_i8_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_7_i8_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i8
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i8 [[VAL_TR]], 7
; CHECK-NEXT:    ret i8 [[TRUNC]]
;
  %shl = shl i64 %val, 7
  %trunc = trunc i64 %shl to i8
  ret i8 %trunc
}

define i2 @trunc_shl_1_i2_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_1_i2_i64(
; CHECK-NEXT:    [[SHL:%.*]] = shl i64 [[VAL:%.*]], 1
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc i64 [[SHL]] to i2
; CHECK-NEXT:    ret i2 [[TRUNC]]
;
  %shl = shl i64 %val, 1
  %trunc = trunc i64 %shl to i2
  ret i2 %trunc
}

define i32 @trunc_shl_1_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_1_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 1
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl i64 %val, 1
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_16_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_16_i32_i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc i64 [[VAL:%.*]] to i32
; CHECK-NEXT:    [[TRUNC:%.*]] = shl i32 [[VAL_TR]], 16
; CHECK-NEXT:    ret i32 [[TRUNC]]
;
  %shl = shl i64 %val, 16
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_33_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_33_i32_i64(
; CHECK-NEXT:    ret i32 0
;
  %shl = shl i64 %val, 33
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

define i32 @trunc_shl_32_i32_i64(i64 %val) {
; CHECK-LABEL: @trunc_shl_32_i32_i64(
; CHECK-NEXT:    ret i32 0
;
  %shl = shl i64 %val, 32
  %trunc = trunc i64 %shl to i32
  ret i32 %trunc
}

; Should be able to handle vectors
define <2 x i32> @trunc_shl_16_v2i32_v2i64(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_16_v2i32_v2i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc <2 x i64> [[VAL:%.*]] to <2 x i32>
; CHECK-NEXT:    [[TRUNC:%.*]] = shl <2 x i32> [[VAL_TR]], splat (i32 16)
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 16, i64 16>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define <2 x i32> @trunc_shl_nosplat_v2i32_v2i64(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_nosplat_v2i32_v2i64(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc <2 x i64> [[VAL:%.*]] to <2 x i32>
; CHECK-NEXT:    [[TRUNC:%.*]] = shl <2 x i32> [[VAL_TR]], <i32 15, i32 16>
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 15, i64 16>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define void @trunc_shl_31_i32_i64_multi_use(i64 %val, ptr addrspace(1) %ptr0, ptr addrspace(1) %ptr1) {
; CHECK-LABEL: @trunc_shl_31_i32_i64_multi_use(
; CHECK-NEXT:    [[SHL:%.*]] = shl i64 [[VAL:%.*]], 31
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc i64 [[SHL]] to i32
; CHECK-NEXT:    store volatile i32 [[TRUNC]], ptr addrspace(1) [[PTR0:%.*]], align 4
; CHECK-NEXT:    store volatile i64 [[SHL]], ptr addrspace(1) [[PTR1:%.*]], align 8
; CHECK-NEXT:    ret void
;
  %shl = shl i64 %val, 31
  %trunc = trunc i64 %shl to i32
  store volatile i32 %trunc, ptr addrspace(1) %ptr0
  store volatile i64 %shl, ptr addrspace(1) %ptr1
  ret void
}

define i32 @trunc_shl_lshr_infloop(i64 %arg) {
; CHECK-LABEL: @trunc_shl_lshr_infloop(
; CHECK-NEXT:    [[ARG_TR:%.*]] = trunc i64 [[ARG:%.*]] to i32
; CHECK-NEXT:    [[TMP1:%.*]] = shl i32 [[ARG_TR]], 1
; CHECK-NEXT:    [[C:%.*]] = and i32 [[TMP1]], -4
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = lshr i64 %arg, 1
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define <2 x i32> @trunc_shl_v2i32_v2i64_uniform(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_v2i32_v2i64_uniform(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc <2 x i64> [[VAL:%.*]] to <2 x i32>
; CHECK-NEXT:    [[TRUNC:%.*]] = shl <2 x i32> [[VAL_TR]], splat (i32 31)
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 31, i64 31>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define <2 x i32> @trunc_shl_v2i32_v2i64_poison(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_v2i32_v2i64_poison(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc <2 x i64> [[VAL:%.*]] to <2 x i32>
; CHECK-NEXT:    [[TRUNC:%.*]] = shl <2 x i32> [[VAL_TR]], <i32 31, i32 poison>
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 31, i64 poison>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define <2 x i32> @trunc_shl_v2i32_v2i64_nonuniform(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_v2i32_v2i64_nonuniform(
; CHECK-NEXT:    [[VAL_TR:%.*]] = trunc <2 x i64> [[VAL:%.*]] to <2 x i32>
; CHECK-NEXT:    [[TRUNC:%.*]] = shl <2 x i32> [[VAL_TR]], <i32 31, i32 12>
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 31, i64 12>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define <2 x i32> @trunc_shl_v2i32_v2i64_outofrange(<2 x i64> %val) {
; CHECK-LABEL: @trunc_shl_v2i32_v2i64_outofrange(
; CHECK-NEXT:    [[SHL:%.*]] = shl <2 x i64> [[VAL:%.*]], <i64 31, i64 33>
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc <2 x i64> [[SHL]] to <2 x i32>
; CHECK-NEXT:    ret <2 x i32> [[TRUNC]]
;
  %shl = shl <2 x i64> %val, <i64 31, i64 33>
  %trunc = trunc <2 x i64> %shl to <2 x i32>
  ret <2 x i32> %trunc
}

define i32 @trunc_shl_ashr_infloop(i64 %arg) {
; CHECK-LABEL: @trunc_shl_ashr_infloop(
; CHECK-NEXT:    [[TMP1:%.*]] = lshr i64 [[ARG:%.*]], 1
; CHECK-NEXT:    [[TMP2:%.*]] = trunc i64 [[TMP1]] to i32
; CHECK-NEXT:    [[C:%.*]] = and i32 [[TMP2]], -4
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = ashr i64 %arg, 3
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define i32 @trunc_shl_shl_infloop(i64 %arg) {
; CHECK-LABEL: @trunc_shl_shl_infloop(
; CHECK-NEXT:    [[ARG_TR:%.*]] = trunc i64 [[ARG:%.*]] to i32
; CHECK-NEXT:    [[C:%.*]] = shl i32 [[ARG_TR]], 3
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = shl i64 %arg, 1
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define i32 @trunc_shl_lshr_var(i64 %arg, i64 %val) {
; CHECK-LABEL: @trunc_shl_lshr_var(
; CHECK-NEXT:    [[A:%.*]] = lshr i64 [[ARG:%.*]], [[VAL:%.*]]
; CHECK-NEXT:    [[A_TR:%.*]] = trunc i64 [[A]] to i32
; CHECK-NEXT:    [[C:%.*]] = shl i32 [[A_TR]], 2
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = lshr i64 %arg, %val
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define i32 @trunc_shl_ashr_var(i64 %arg, i64 %val) {
; CHECK-LABEL: @trunc_shl_ashr_var(
; CHECK-NEXT:    [[A:%.*]] = ashr i64 [[ARG:%.*]], [[VAL:%.*]]
; CHECK-NEXT:    [[A_TR:%.*]] = trunc i64 [[A]] to i32
; CHECK-NEXT:    [[C:%.*]] = shl i32 [[A_TR]], 2
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = ashr i64 %arg, %val
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define i32 @trunc_shl_shl_var(i64 %arg, i64 %val) {
; CHECK-LABEL: @trunc_shl_shl_var(
; CHECK-NEXT:    [[A:%.*]] = shl i64 [[ARG:%.*]], [[VAL:%.*]]
; CHECK-NEXT:    [[A_TR:%.*]] = trunc i64 [[A]] to i32
; CHECK-NEXT:    [[C:%.*]] = shl i32 [[A_TR]], 2
; CHECK-NEXT:    ret i32 [[C]]
;
  %A = shl i64 %arg, %val
  %B = shl i64 %A, 2
  %C = trunc i64 %B to i32
  ret i32 %C
}

define <8 x i16> @trunc_shl_v8i15_v8i32_15(<8 x i32> %a) {
; CHECK-LABEL: @trunc_shl_v8i15_v8i32_15(
; CHECK-NEXT:    [[A_TR:%.*]] = trunc <8 x i32> [[A:%.*]] to <8 x i16>
; CHECK-NEXT:    [[CONV:%.*]] = shl <8 x i16> [[A_TR]], splat (i16 15)
; CHECK-NEXT:    ret <8 x i16> [[CONV]]
;
  %shl = shl <8 x i32> %a, <i32 15, i32 15, i32 15, i32 15, i32 15, i32 15, i32 15, i32 15>
  %conv = trunc <8 x i32> %shl to <8 x i16>
  ret <8 x i16> %conv
}

define <8 x i16> @trunc_shl_v8i16_v8i32_16(<8 x i32> %a) {
; CHECK-LABEL: @trunc_shl_v8i16_v8i32_16(
; CHECK-NEXT:    ret <8 x i16> zeroinitializer
;
  %shl = shl <8 x i32> %a, <i32 16, i32 16, i32 16, i32 16, i32 16, i32 16, i32 16, i32 16>
  %conv = trunc <8 x i32> %shl to <8 x i16>
  ret <8 x i16> %conv
}

define <8 x i16> @trunc_shl_v8i16_v8i32_17(<8 x i32> %a) {
; CHECK-LABEL: @trunc_shl_v8i16_v8i32_17(
; CHECK-NEXT:    ret <8 x i16> zeroinitializer
;
  %shl = shl <8 x i32> %a, <i32 17, i32 17, i32 17, i32 17, i32 17, i32 17, i32 17, i32 17>
  %conv = trunc <8 x i32> %shl to <8 x i16>
  ret <8 x i16> %conv
}

define <8 x i16> @trunc_shl_v8i16_v8i32_4(<8 x i32> %a) {
; CHECK-LABEL: @trunc_shl_v8i16_v8i32_4(
; CHECK-NEXT:    [[A_TR:%.*]] = trunc <8 x i32> [[A:%.*]] to <8 x i16>
; CHECK-NEXT:    [[CONV:%.*]] = shl <8 x i16> [[A_TR]], splat (i16 4)
; CHECK-NEXT:    ret <8 x i16> [[CONV]]
;
  %shl = shl <8 x i32> %a, <i32 4, i32 4, i32 4, i32 4, i32 4, i32 4, i32 4, i32 4>
  %conv = trunc <8 x i32> %shl to <8 x i16>
  ret <8 x i16> %conv
}

; Although the mask is the same value, we don't create a shuffle for types that the backend may not be able to handle:
; trunc (shuffle X, C, Mask) --> shuffle (trunc X), C', Mask

define <4 x i8> @wide_shuf(<4 x i32> %x) {
; CHECK-LABEL: @wide_shuf(
; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <4 x i32> [[X:%.*]], <4 x i32> <i32 poison, i32 3634, i32 90, i32 poison>, <4 x i32> <i32 1, i32 5, i32 6, i32 2>
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc <4 x i32> [[SHUF]] to <4 x i8>
; CHECK-NEXT:    ret <4 x i8> [[TRUNC]]
;
  %shuf = shufflevector <4 x i32> %x, <4 x i32> <i32 35, i32 3634, i32 90, i32 -1>, <4 x i32> <i32 1, i32 5, i32 6, i32 2>
  %trunc = trunc <4 x i32> %shuf to <4 x i8>
  ret <4 x i8> %trunc
}

; trunc (shuffle X, poison, SplatMask) --> shuffle (trunc X), poison, SplatMask

define <4 x i8> @wide_splat1(<4 x i32> %x) {
; CHECK-LABEL: @wide_splat1(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <4 x i32> [[X:%.*]] to <4 x i8>
; CHECK-NEXT:    [[TRUNC:%.*]] = shufflevector <4 x i8> [[TMP1]], <4 x i8> poison, <4 x i32> <i32 2, i32 2, i32 2, i32 2>
; CHECK-NEXT:    ret <4 x i8> [[TRUNC]]
;
  %shuf = shufflevector <4 x i32> %x, <4 x i32> poison, <4 x i32> <i32 2, i32 2, i32 2, i32 2>
  %trunc = trunc <4 x i32> %shuf to <4 x i8>
  ret <4 x i8> %trunc
}

; Test weird types.
; trunc (shuffle X, poison, SplatMask) --> shuffle (trunc X), poison, SplatMask

define <3 x i31> @wide_splat2(<3 x i33> %x) {
; CHECK-LABEL: @wide_splat2(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <3 x i33> [[X:%.*]] to <3 x i31>
; CHECK-NEXT:    [[TRUNC:%.*]] = shufflevector <3 x i31> [[TMP1]], <3 x i31> poison, <3 x i32> <i32 1, i32 1, i32 1>
; CHECK-NEXT:    ret <3 x i31> [[TRUNC]]
;
  %shuf = shufflevector <3 x i33> %x, <3 x i33> poison, <3 x i32> <i32 1, i32 1, i32 1>
  %trunc = trunc <3 x i33> %shuf to <3 x i31>
  ret <3 x i31> %trunc
}

; FIXME:
; trunc (shuffle X, poison, SplatMask) --> shuffle (trunc X), poison, SplatMask
; A mask with poison elements should still be considered a splat mask.

define <3 x i31> @wide_splat3(<3 x i33> %x) {
; CHECK-LABEL: @wide_splat3(
; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <3 x i33> [[X:%.*]], <3 x i33> poison, <3 x i32> <i32 poison, i32 1, i32 1>
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc <3 x i33> [[SHUF]] to <3 x i31>
; CHECK-NEXT:    ret <3 x i31> [[TRUNC]]
;
  %shuf = shufflevector <3 x i33> %x, <3 x i33> poison, <3 x i32> <i32 poison, i32 1, i32 1>
  %trunc = trunc <3 x i33> %shuf to <3 x i31>
  ret <3 x i31> %trunc
}

; TODO: The shuffle extends the length of the input vector. Should we shrink this?

define <8 x i8> @wide_lengthening_splat(<4 x i16> %v) {
; CHECK-LABEL: @wide_lengthening_splat(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <4 x i16> [[V:%.*]] to <4 x i8>
; CHECK-NEXT:    [[TR:%.*]] = shufflevector <4 x i8> [[TMP1]], <4 x i8> poison, <8 x i32> zeroinitializer
; CHECK-NEXT:    ret <8 x i8> [[TR]]
;
  %shuf = shufflevector <4 x i16> %v, <4 x i16> %v, <8 x i32> zeroinitializer
  %tr = trunc <8 x i16> %shuf to <8 x i8>
  ret <8 x i8> %tr
}

; This is a negative test, we expect the trunc to remain after the shuffle as it
; might not be beneficial to preform trunc on a wider type
define <4 x i8> @wide_shortening_splat(<8 x i16> %v) {
; CHECK-LABEL: @wide_shortening_splat(
; CHECK-NEXT:    [[SHUF:%.*]] = shufflevector <8 x i16> [[V:%.*]], <8 x i16> poison, <4 x i32> zeroinitializer
; CHECK-NEXT:    [[TR:%.*]] = trunc <4 x i16> [[SHUF]] to <4 x i8>
; CHECK-NEXT:    ret <4 x i8> [[TR]]
;
  %shuf = shufflevector <8 x i16> %v, <8 x i16> %v, <4 x i32> zeroinitializer
  %tr = trunc <4 x i16> %shuf to <4 x i8>
  ret <4 x i8> %tr
}

define <2 x i8> @narrow_add_vec_constant(<2 x i32> %x) {
; CHECK-LABEL: @narrow_add_vec_constant(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <2 x i32> [[X:%.*]] to <2 x i8>
; CHECK-NEXT:    [[TR:%.*]] = add <2 x i8> [[TMP1]], <i8 0, i8 127>
; CHECK-NEXT:    ret <2 x i8> [[TR]]
;
  %add = add <2 x i32> %x, <i32 256, i32 -129>
  %tr = trunc <2 x i32> %add to <2 x i8>
  ret <2 x i8> %tr
}

define <2 x i8> @narrow_mul_vec_constant(<2 x i32> %x) {
; CHECK-LABEL: @narrow_mul_vec_constant(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <2 x i32> [[X:%.*]] to <2 x i8>
; CHECK-NEXT:    [[TR:%.*]] = mul <2 x i8> [[TMP1]], <i8 0, i8 127>
; CHECK-NEXT:    ret <2 x i8> [[TR]]
;
  %add = mul <2 x i32> %x, <i32 256, i32 -129>
  %tr = trunc <2 x i32> %add to <2 x i8>
  ret <2 x i8> %tr
}

define <2 x i8> @narrow_sub_vec_constant(<2 x i32> %x) {
; CHECK-LABEL: @narrow_sub_vec_constant(
; CHECK-NEXT:    [[TMP1:%.*]] = trunc <2 x i32> [[X:%.*]] to <2 x i8>
; CHECK-NEXT:    [[TR:%.*]] = sub <2 x i8> <i8 0, i8 127>, [[TMP1]]
; CHECK-NEXT:    ret <2 x i8> [[TR]]
;
  %sub = sub <2 x i32> <i32 256, i32 -129>, %x
  %tr = trunc <2 x i32> %sub to <2 x i8>
  ret <2 x i8> %tr
}

; If the select is narrowed based on the target's datalayout, we allow more optimizations.

define i16 @PR44545(i32 %t0, i32 %data) {
; CHECK-LABEL: @PR44545(
; CHECK-NEXT:    [[ISZERO:%.*]] = icmp eq i32 [[DATA:%.*]], 0
; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[T0:%.*]] to i16
; CHECK-NEXT:    [[SUB:%.*]] = select i1 [[ISZERO]], i16 -1, i16 [[TMP1]]
; CHECK-NEXT:    ret i16 [[SUB]]
;
  %t1 = add nuw nsw i32 %t0, 1
  %iszero = icmp eq i32 %data, 0
  %ffs = select i1 %iszero, i32 0, i32 %t1
  %cast = trunc i32 %ffs to i16
  %sub = add nsw i16 %cast, -1
  ret i16 %sub
}

; Make sure that SimplifyDemandedBits drops the nowrap flags
define i8 @drop_nsw_trunc(i16 %x, i16 %y) {
; CHECK-LABEL: @drop_nsw_trunc(
; CHECK-NEXT:    [[AND2:%.*]] = and i16 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    [[RES:%.*]] = trunc i16 [[AND2]] to i8
; CHECK-NEXT:    ret i8 [[RES]]
;
  %and = and i16 %x, 255
  %and2 = and i16 %and, %y
  %res = trunc nsw i16 %and2 to i8
  ret i8 %res
}

define i8 @drop_nuw_trunc(i16 %x, i16 %y) {
; CHECK-LABEL: @drop_nuw_trunc(
; CHECK-NEXT:    [[AND2:%.*]] = and i16 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    [[RES:%.*]] = trunc i16 [[AND2]] to i8
; CHECK-NEXT:    ret i8 [[RES]]
;
  %and = and i16 %x, 255
  %and2 = and i16 %and, %y
  %res = trunc nuw i16 %and2 to i8
  ret i8 %res
}

define i8 @drop_both_trunc(i16 %x, i16 %y) {
; CHECK-LABEL: @drop_both_trunc(
; CHECK-NEXT:    [[AND2:%.*]] = and i16 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    [[RES:%.*]] = trunc i16 [[AND2]] to i8
; CHECK-NEXT:    ret i8 [[RES]]
;
  %and = and i16 %x, 255
  %and2 = and i16 %and, %y
  %res = trunc nuw nsw i16 %and2 to i8
  ret i8 %res
}

define i1 @trunc_xor(i8 %x, i8 %y) {
; CHECK-LABEL: @trunc_xor(
; CHECK-NEXT:    [[XOR:%.*]] = xor i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    [[R:%.*]] = trunc i8 [[XOR]] to i1
; CHECK-NEXT:    ret i1 [[R]]
;
  %xor = xor i8 %x, %y
  %r = trunc i8 %xor to i1
  ret i1 %r
}

define i1 @trunc_nuw_xor(i8 %x, i8 %y) {
; CHECK-LABEL: @trunc_nuw_xor(
; CHECK-NEXT:    [[R:%.*]] = icmp ne i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    ret i1 [[R]]
;
  %xor = xor i8 %x, %y
  %r = trunc nuw i8 %xor to i1
  ret i1 %r
}

define i1 @trunc_nsw_xor(i8 %x, i8 %y) {
; CHECK-LABEL: @trunc_nsw_xor(
; CHECK-NEXT:    [[R:%.*]] = icmp ne i8 [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    ret i1 [[R]]
;
  %xor = xor i8 %x, %y
  %r = trunc nsw i8 %xor to i1
  ret i1 %r
}

define <2 x i1> @trunc_nuw_xor_vector(<2 x i8> %x, <2 x i8> %y) {
; CHECK-LABEL: @trunc_nuw_xor_vector(
; CHECK-NEXT:    [[R:%.*]] = icmp ne <2 x i8> [[X:%.*]], [[Y:%.*]]
; CHECK-NEXT:    ret <2 x i1> [[R]]
;
  %xor = xor <2 x i8> %x, %y
  %r = trunc nuw <2 x i8> %xor to <2 x i1>
  ret <2 x i1> %r
}

define void @pr95547(i32 %x) {
; CHECK-LABEL: @pr95547(
; CHECK-NEXT:    [[X_TRUNC:%.*]] = trunc i32 [[X:%.*]] to i16
; CHECK-NEXT:    [[DIV:%.*]] = udiv i16 11, [[X_TRUNC]]
; CHECK-NEXT:    [[CMP:%.*]] = icmp ult i32 [[X]], 256
; CHECK-NEXT:    br i1 [[CMP]], label [[LOOP:%.*]], label [[EXIT:%.*]]
; CHECK:       loop:
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc nuw nsw i16 [[DIV]] to i8
; CHECK-NEXT:    call void @use.i8(i8 [[TRUNC]])
; CHECK-NEXT:    br label [[LOOP]]
; CHECK:       exit:
; CHECK-NEXT:    ret void
;
  %x.trunc = trunc i32 %x to i16
  %div = udiv i16 11, %x.trunc
  %cmp = icmp ult i32 %x, 256
  br i1 %cmp, label %loop, label %exit

loop:
  ; The loop is just here to prevent sinking.
  %trunc = trunc i16 %div to i8
  call void @use.i8(i8 %trunc)
  br label %loop

exit:
  ret void
}

define i1 @trunc_nuw_i1_non_zero(i8 %1) {
; CHECK-LABEL: @trunc_nuw_i1_non_zero(
; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i8 [[TMP0:%.*]], 0
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[TMP2]])
; CHECK-NEXT:    ret i1 true
;
  %3 = icmp ne i8 %1, 0
  tail call void @llvm.assume(i1 %3)
  %ret = trunc nuw i8 %1 to i1
  ret i1 %ret
}

define i1 @neg_trunc_nuw_i1_maybe_zero(i8 %1) {
; CHECK-LABEL: @neg_trunc_nuw_i1_maybe_zero(
; CHECK-NEXT:    [[RET:%.*]] = trunc nuw i8 [[TMP0:%.*]] to i1
; CHECK-NEXT:    ret i1 [[RET]]
;
  %ret = trunc nuw i8 %1 to i1
  ret i1 %ret
}

define i2 @neg_trunc_nuw_i2_non_zero(i8 %1) {
; CHECK-LABEL: @neg_trunc_nuw_i2_non_zero(
; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i8 [[TMP0:%.*]], 0
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[TMP2]])
; CHECK-NEXT:    [[RET:%.*]] = trunc nuw i8 [[TMP0]] to i2
; CHECK-NEXT:    ret i2 [[RET]]
;
  %3 = icmp ne i8 %1, 0
  tail call void @llvm.assume(i1 %3)
  %ret = trunc nuw i8 %1 to i2
  ret i2 %ret
}

define i1 @neg_trunc_i1_non_zero(i8 %1) {
; CHECK-LABEL: @neg_trunc_i1_non_zero(
; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i8 [[TMP0:%.*]], 0
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[TMP2]])
; CHECK-NEXT:    [[RET:%.*]] = trunc i8 [[TMP0]] to i1
; CHECK-NEXT:    ret i1 [[RET]]
;
  %3 = icmp ne i8 %1, 0
  tail call void @llvm.assume(i1 %3)
  %ret = trunc i8 %1 to i1
  ret i1 %ret
}

define i1 @trunc_nsw_i1_non_zero(i8 %1) {
; CHECK-LABEL: @trunc_nsw_i1_non_zero(
; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i8 [[TMP0:%.*]], 0
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[TMP2]])
; CHECK-NEXT:    ret i1 true
;
  %3 = icmp ne i8 %1, 0
  tail call void @llvm.assume(i1 %3)
  %ret = trunc nsw i8 %1 to i1
  ret i1 %ret
}

define i1 @neg_trunc_nsw_i1_maybe_zero(i8 %1) {
; CHECK-LABEL: @neg_trunc_nsw_i1_maybe_zero(
; CHECK-NEXT:    [[RET:%.*]] = trunc nsw i8 [[TMP0:%.*]] to i1
; CHECK-NEXT:    ret i1 [[RET]]
;
  %ret = trunc nsw i8 %1 to i1
  ret i1 %ret
}

define i2 @neg_trunc_nsw_i2_non_zero(i8 %1) {
; CHECK-LABEL: @neg_trunc_nsw_i2_non_zero(
; CHECK-NEXT:    [[TMP2:%.*]] = icmp ne i8 [[TMP0:%.*]], 0
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[TMP2]])
; CHECK-NEXT:    [[RET:%.*]] = trunc nsw i8 [[TMP0]] to i2
; CHECK-NEXT:    ret i2 [[RET]]
;
  %3 = icmp ne i8 %1, 0
  tail call void @llvm.assume(i1 %3)
  %ret = trunc nsw i8 %1 to i2
  ret i2 %ret
}

define i1 @trunc_i1_usub_sat_one(i8 %x) {
; CHECK-LABEL: @trunc_i1_usub_sat_one(
; CHECK-NEXT:    [[TRUNC:%.*]] = icmp eq i8 [[X:%.*]], 0
; CHECK-NEXT:    ret i1 [[TRUNC]]
;
  %call = call i8 @llvm.usub.sat.i8(i8 1, i8 %x)
  %trunc = trunc i8 %call to i1
  ret i1 %trunc
}

define <2 x i1> @trunc_i1_usub_sat_one_vec(<2 x i8> %x) {
; CHECK-LABEL: @trunc_i1_usub_sat_one_vec(
; CHECK-NEXT:    [[TRUNC:%.*]] = icmp eq <2 x i8> [[X:%.*]], zeroinitializer
; CHECK-NEXT:    ret <2 x i1> [[TRUNC]]
;
  %call = call <2 x i8> @llvm.usub.sat.v2i8(<2 x i8> <i8 1, i8 1>, <2 x i8> %x)
  %trunc = trunc <2 x i8> %call to <2 x i1>
  ret <2 x i1> %trunc
}

define i1 @neg_trunc_i1_usub_sat_two(i8 %x) {
; CHECK-LABEL: @neg_trunc_i1_usub_sat_two(
; CHECK-NEXT:    [[CALL:%.*]] = call i8 @llvm.usub.sat.i8(i8 2, i8 [[X:%.*]])
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc i8 [[CALL]] to i1
; CHECK-NEXT:    ret i1 [[TRUNC]]
;
  %call = call i8 @llvm.usub.sat.i8(i8 2, i8 %x)
  %trunc = trunc i8 %call to i1
  ret i1 %trunc
}

define i2 @neg_trunc_i2_usub_sat_one(i8 %x) {
; CHECK-LABEL: @neg_trunc_i2_usub_sat_one(
; CHECK-NEXT:    [[CALL:%.*]] = call i8 @llvm.usub.sat.i8(i8 1, i8 [[X:%.*]])
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc nuw nsw i8 [[CALL]] to i2
; CHECK-NEXT:    ret i2 [[TRUNC]]
;
  %call = call i8 @llvm.usub.sat.i8(i8 1, i8 %x)
  %trunc = trunc i8 %call to i2
  ret i2 %trunc
}

define i1 @neg_trunc_i1_usub_sat_one_multi_use(i8 %x) {
; CHECK-LABEL: @neg_trunc_i1_usub_sat_one_multi_use(
; CHECK-NEXT:    [[CALL:%.*]] = call i8 @llvm.usub.sat.i8(i8 1, i8 [[X:%.*]])
; CHECK-NEXT:    [[TRUNC:%.*]] = trunc nuw i8 [[CALL]] to i1
; CHECK-NEXT:    call void @use.i8(i8 [[CALL]])
; CHECK-NEXT:    ret i1 [[TRUNC]]
;
  %call = call i8 @llvm.usub.sat.i8(i8 1, i8 %x)
  %trunc = trunc i8 %call to i1
  call void @use.i8(i8 %call)
  ret i1 %trunc
}

define i32 @zext_i32_trunc_nuw_i8(i16 %x, i32 %y) {
; CHECK-LABEL: @zext_i32_trunc_nuw_i8(
; CHECK-NEXT:    [[TRUNC1:%.*]] = zext nneg i16 [[X:%.*]] to i32
; CHECK-NEXT:    [[OR:%.*]] = or i32 [[Y:%.*]], [[TRUNC1]]
; CHECK-NEXT:    [[E:%.*]] = and i32 [[OR]], 255
; CHECK-NEXT:    ret i32 [[E]]
;
  %trunc1 = trunc nuw i16 %x to i8
  %trunc2 = trunc i32 %y to i8
  %or = or i8 %trunc1, %trunc2
  %e = zext i8 %or to i32
  ret i32 %e
}

define i32 @neg_zext_i32_trunc_nsw_i8(i16 %x, i32 %y) {
; CHECK-LABEL: @neg_zext_i32_trunc_nsw_i8(
; CHECK-NEXT:    [[TRUNC1:%.*]] = zext i16 [[X:%.*]] to i32
; CHECK-NEXT:    [[OR:%.*]] = or i32 [[Y:%.*]], [[TRUNC1]]
; CHECK-NEXT:    [[E:%.*]] = and i32 [[OR]], 255
; CHECK-NEXT:    ret i32 [[E]]
;
  %trunc1 = trunc nsw i16 %x to i8
  %trunc2 = trunc i32 %y to i8
  %or = or i8 %trunc1, %trunc2
  %e = zext i8 %or to i32
  ret i32 %e
}

define i16 @zext_i16_trunc_nuw_nsw_i8(i32 %x) {
; CHECK-LABEL: @zext_i16_trunc_nuw_nsw_i8(
; CHECK-NEXT:    [[C:%.*]] = trunc nuw nsw i32 [[X:%.*]] to i16
; CHECK-NEXT:    [[E:%.*]] = and i16 [[C]], 255
; CHECK-NEXT:    ret i16 [[E]]
;
  %c = trunc nuw nsw i32 %x to i8
  %e = zext i8 %c to i16
  ret i16 %e
}

define i16 @zext_i16_trunc_nsw_i8(i32 %x) {
; CHECK-LABEL: @zext_i16_trunc_nsw_i8(
; CHECK-NEXT:    [[C:%.*]] = trunc nsw i32 [[X:%.*]] to i16
; CHECK-NEXT:    [[E:%.*]] = and i16 [[C]], 255
; CHECK-NEXT:    ret i16 [[E]]
;
  %c = trunc nsw i32 %x to i8
  %e = zext i8 %c to i16
  ret i16 %e
}

define i16 @zext_i16_trunc_nuw_i8(i32 %x) {
; CHECK-LABEL: @zext_i16_trunc_nuw_i8(
; CHECK-NEXT:    [[C:%.*]] = trunc nuw i32 [[X:%.*]] to i16
; CHECK-NEXT:    [[E:%.*]] = and i16 [[C]], 255
; CHECK-NEXT:    ret i16 [[E]]
;
  %c = trunc nuw i32 %x to i8
  %e = zext i8 %c to i16
  ret i16 %e
}

define i8 @neg_trunc_i8_trunc_nuw_i16(i32 %x) {
; CHECK-LABEL: @neg_trunc_i8_trunc_nuw_i16(
; CHECK-NEXT:    [[E:%.*]] = trunc i32 [[X:%.*]] to i8
; CHECK-NEXT:    ret i8 [[E]]
;
  %c = trunc nuw i32 %x to i16
  %e = trunc i16 %c to i8
  ret i8 %e
}

define i8 @neg_trunc_nuw_i8_trunc_nuw_i16(i32 %x) {
; CHECK-LABEL: @neg_trunc_nuw_i8_trunc_nuw_i16(
; CHECK-NEXT:    [[E:%.*]] = trunc i32 [[X:%.*]] to i8
; CHECK-NEXT:    ret i8 [[E]]
;
  %c = trunc nuw i32 %x to i16
  %e = trunc nuw i16 %c to i8
  ret i8 %e
}

define i1 @trunc_nuw_lshr_exact(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nuw_lshr_exact(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X:%.*]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %lshr = lshr exact i8 %x, %c
  %ret = trunc nuw i8 %lshr to i1
  ret i1 %ret
}

define i1 @trunc_nsw_lshr_exact(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nsw_lshr_exact(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X:%.*]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %lshr = lshr exact i8 %x, %c
  %ret = trunc nsw i8 %lshr to i1
  ret i1 %ret
}

define i1 @trunc_nuw_ashr_exact(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nuw_ashr_exact(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X:%.*]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %ashr = ashr exact i8 %x, %c
  %ret = trunc nuw i8 %ashr to i1
  ret i1 %ret
}

define i1 @trunc_nsw_ashr_exact(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nsw_ashr_exact(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X:%.*]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %ashr = ashr exact i8 %x, %c
  %ret = trunc nsw i8 %ashr to i1
  ret i1 %ret
}

define i1 @trunc_nuw_lshr_exact_use(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nuw_lshr_exact_use(
; CHECK-NEXT:    [[LSHR:%.*]] = lshr exact i8 [[X:%.*]], [[C:%.*]]
; CHECK-NEXT:    call void @use.i8(i8 [[LSHR]])
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %lshr = lshr exact i8 %x, %c
  call void @use.i8(i8 %lshr)
  %ret = trunc nuw i8 %lshr to i1
  ret i1 %ret
}

define i1 @trunc_nuw_ashr_exact_use(i8 %x, i8 %c) {
; CHECK-LABEL: @trunc_nuw_ashr_exact_use(
; CHECK-NEXT:    [[ASHR:%.*]] = ashr exact i8 [[X:%.*]], [[C:%.*]]
; CHECK-NEXT:    call void @use.i8(i8 [[ASHR]])
; CHECK-NEXT:    [[RET:%.*]] = icmp ne i8 [[X]], 0
; CHECK-NEXT:    ret i1 [[RET]]
;
  %ashr = ashr exact i8 %x, %c
  call void @use.i8(i8 %ashr)
  %ret = trunc nuw i8 %ashr to i1
  ret i1 %ret
}

define <2 x i1> @trunc_nuw_lshr_icmp_vec(<2 x i8> %x, <2 x i8> %c) {
; CHECK-LABEL: @trunc_nuw_lshr_icmp_vec(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne <2 x i8> [[X:%.*]], zeroinitializer
; CHECK-NEXT:    ret <2 x i1> [[RET]]
;
  %lshr = lshr exact <2 x i8> %x, %c
  %ret = trunc nuw <2 x i8> %lshr to <2 x i1>
  ret <2 x i1> %ret
}

define <2 x i1> @trunc_nuw_ashr_icmp_vec(<2 x i8> %x, <2 x i8> %c) {
; CHECK-LABEL: @trunc_nuw_ashr_icmp_vec(
; CHECK-NEXT:    [[RET:%.*]] = icmp ne <2 x i8> [[X:%.*]], zeroinitializer
; CHECK-NEXT:    ret <2 x i1> [[RET]]
;
  %ashr = ashr exact <2 x i8> %x, %c
  %ret = trunc nuw <2 x i8> %ashr to <2 x i1>
  ret <2 x i1> %ret
}

define i1 @neg_trunc_lshr_exact(i8 %x, i8 %c) {
; CHECK-LABEL: @neg_trunc_lshr_exact(
; CHECK-NEXT:    [[LSHR:%.*]] = lshr exact i8 [[X:%.*]], [[C:%.*]]
; CHECK-NEXT:    [[RET:%.*]] = trunc i8 [[LSHR]] to i1
; CHECK-NEXT:    ret i1 [[RET]]
;
  %lshr = lshr exact i8 %x, %c
  %ret = trunc i8 %lshr to i1
  ret i1 %ret
}

define i1 @neg_trunc_nuw_lshr(i8 %x, i8 %c) {
; CHECK-LABEL: @neg_trunc_nuw_lshr(
; CHECK-NEXT:    [[LSHR:%.*]] = lshr i8 [[X:%.*]], [[C:%.*]]
; CHECK-NEXT:    [[RET:%.*]] = trunc nuw i8 [[LSHR]] to i1
; CHECK-NEXT:    ret i1 [[RET]]
;
  %lshr = lshr i8 %x, %c
  %ret = trunc nuw i8 %lshr to i1
  ret i1 %ret
}

define i32 @separate_truncs_i32(i64 %x) {
; CHECK-LABEL: @separate_truncs_i32(
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr i64 %x, 16
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc i64 %[[SHIFT]] to i32
; CHECK-NEXT:    %[[MASK:.*]] = and i32 %[[TRUNC]], 130816
; CHECK-NEXT:    ret i32 %[[MASK]]
;
  %wide.shift = lshr i64 %x, 16
  %wide = trunc i64 %wide.shift to i32
  %field.hi = and i32 %wide, 65536

  %x.narrow = trunc i64 %x to i32
  %narrow.shift = lshr i32 %x.narrow, 16
  %field.lo = and i32 %narrow.shift, 65280

  %result = or i32 %field.hi, %field.lo
  ret i32 %result
}

define i32 @separate_truncs_i32_reverse(i64 %x) {
; CHECK-LABEL: @separate_truncs_i32_reverse(
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr i64 %x, 16
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc i64 %[[SHIFT]] to i32
; CHECK-NEXT:    %[[MASK:.*]] = and i32 %[[TRUNC]], 130816
; CHECK-NEXT:    ret i32 %[[MASK]]
;
  %x.narrow = trunc i64 %x to i32
  %narrow.shift = lshr i32 %x.narrow, 16
  %field.lo = and i32 %narrow.shift, 65280

  %wide.shift = lshr i64 %x, 16 
  %wide = trunc i64 %wide.shift to i32
  %field.hi = and i32 %wide, 65536

  %result = or i32 %field.lo, %field.hi
  ret i32 %result
}

define i32 @separate_truncs_i32_multiuse(i64 %x) {
; CHECK-LABEL: @separate_truncs_i32_multiuse(
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr i64 %x, 16
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc i64 %[[SHIFT]] to i32
; CHECK-NEXT:    call void @use(i32 %[[TRUNC]])
; CHECK-NEXT:    %[[MASK:.*]] = and i32 %[[TRUNC]], 130816
; CHECK-NEXT:    ret i32 %[[MASK]]
; 
  %wide.shift = lshr i64 %x, 16
  %wide = trunc i64 %wide.shift to i32
  call void @use(i32 %wide)
  %field.hi = and i32 %wide, 65536

  %x.narrow = trunc i64 %x to i32
  %narrow.shift = lshr i32 %x.narrow, 16
  %field.lo = and i32 %narrow.shift, 65280

  %result = or i32 %field.hi, %field.lo
  ret i32 %result
}

define i32 @separate_truncs_i32_reverse_multiuse(i64 %x) {
; CHECK-LABEL: @separate_truncs_i32_reverse_multiuse(
; CHECK-NEXT:    %[[NARROW:.*]] = trunc i64 %x to i32
; CHECK-NEXT:    call void @use(i32 %[[NARROW]])
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr i64 %x, 16
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc i64 %[[SHIFT]] to i32
; CHECK-NEXT:    %[[MASK:.*]] = and i32 %[[TRUNC]], 130816
; CHECK-NEXT:    ret i32 %[[MASK]]
;
  %x.narrow = trunc i64 %x to i32
  call void @use(i32 %x.narrow)
  %narrow.shift = lshr i32 %x.narrow, 16
  %field.lo = and i32 %narrow.shift, 65280

  %wide.shift = lshr i64 %x, 16 
  %wide = trunc i64 %wide.shift to i32
  %field.hi = and i32 %wide, 65536

  %result = or i32 %field.lo, %field.hi
  ret i32 %result
}

define <2 x i32> @separate_truncs_i32_reverse_vec(<2 x i64> %x) {
; CHECK-LABEL: @separate_truncs_i32_reverse_vec(
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr <2 x i64> %x, splat (i64 16)
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc <2 x i64> %[[SHIFT]] to <2 x i32>
; CHECK-NEXT:    %[[MASK:.*]] = and <2 x i32> %[[TRUNC]], splat (i32 130816)
; CHECK-NEXT:    ret <2 x i32> %[[MASK]]
; 
  %x.narrow = trunc <2 x i64> %x to <2 x i32>
  %narrow.shift = lshr <2 x i32> %x.narrow, <i32 16, i32 16>
  %field.lo = and <2 x i32> %narrow.shift, <i32 65280, i32 65280>

  %wide.shift = lshr <2 x i64> %x, <i64 16, i64 16>
  %wide = trunc <2 x i64> %wide.shift to <2 x i32>
  %field.hi = and <2 x i32> %wide, <i32 65536, i32 65536>

  %result = or <2 x i32> %field.lo, %field.hi
  ret <2 x i32> %result
}

define i32 @neg_separate_truncs_i32_shift_geq_32(i64 %x) {
; CHECK-LABEL: @neg_separate_truncs_i32_shift_geq_32(
; CHECK-NEXT:    ret i32 poison
;
  %wide.shift = lshr i64 %x, 32
  %wide = trunc i64 %wide.shift to i32
  %field.hi = and i32 %wide, 65536

  %x.narrow = trunc i64 %x to i32
  %narrow.shift = lshr i32 %x.narrow, 32
  %field.lo = and i32 %narrow.shift, 65280

  %result = or i32 %field.hi, %field.lo
  ret i32 %result
}

; negative test: narrow mask 131072
define i32 @neg_separate_truncs_i32_upper_bits(i64 %x) {
; CHECK-LABEL: @neg_separate_truncs_i32_upper_bits(
; CHECK-NEXT:    %[[SHIFT:.*]] = lshr i64 %x, 16
; CHECK-NEXT:    %[[TRUNC:.*]] = trunc i64 %[[SHIFT]] to i32
; CHECK-NEXT:    %[[MASK:.*]] = and i32 %[[TRUNC]], 65536
; CHECK-NEXT:    ret i32 %[[MASK]]
;
  %wide.shift = lshr i64 %x, 16
  %wide = trunc i64 %wide.shift to i32
  %field.hi = and i32 %wide, 65536

  %x.narrow = trunc i64 %x to i32
  %narrow.shift = lshr i32 %x.narrow, 16
  %field.lo = and i32 %narrow.shift, 131072

  %result = or i32 %field.hi, %field.lo
  ret i32 %result
}