; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5

; RUN: opt -S -passes=instcombine < %s | FileCheck %s

; When C0 is neither 0 nor 1:
;   umax(nuw_mul(x, C0), x + 1) is optimized to:
;   x == 0 ? 1 : nuw_mul(x, C0)
; When C0 is not 0:
;   umax(nuw_shl(x, C0), x + 1) is optimized to:
;   x == 0 ? 1 : nuw_shl(x, C0)

; Positive Test Cases for `shl`

define i64 @test_shl_by_2(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_by_2(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[TMP2:%.*]] = shl nuw i64 [[X]], 2
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[TMP2]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 2
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_by_5(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_by_5(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[TMP2:%.*]] = shl nuw i64 [[X]], 5
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[TMP2]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 5
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_with_nsw(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_with_nsw(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw nsw i64 [[X]], 2
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[SHL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw nsw i64 %x, 2
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define <2 x i64> @test_shl_vector_by_2(<2 x i64> %x) {
; CHECK-LABEL: define <2 x i64> @test_shl_vector_by_2(
; CHECK-SAME: <2 x i64> [[X:%.*]]) {
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw <2 x i64> [[X]], splat (i64 2)
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq <2 x i64> [[X]], zeroinitializer
; CHECK-NEXT:    [[MAX:%.*]] = select <2 x i1> [[TMP1]], <2 x i64> splat (i64 1), <2 x i64> [[SHL]]
; CHECK-NEXT:    ret <2 x i64> [[MAX]]
;
  %x1 = add <2 x i64> %x, <i64 1, i64 1>
  %shl = shl nuw <2 x i64> %x, <i64 2, i64 2>
  %max = call <2 x i64> @llvm.umax.v2i64(<2 x i64> %shl, <2 x i64> %x1)
  ret <2 x i64> %max
}

; Commuted Test Cases for `shl`

define i64 @test_shl_umax_commuted(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_umax_commuted(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw i64 [[X]], 2
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[SHL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 2
  %max = call i64 @llvm.umax.i64(i64 %x1, i64 %shl)
  ret i64 %max
}

; Negative Test Cases for `shl`

define i64 @test_shl_by_zero(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_by_zero(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[X]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 0
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_add_by_2(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_add_by_2(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 2
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw i64 [[X]], 2
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[SHL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 2
  %shl = shl nuw i64 %x, 2
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_without_nuw(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_without_nuw(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[SHL:%.*]] = shl i64 [[X]], 2
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[SHL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl i64 %x, 2
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_umin(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_umin(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw i64 [[X]], 2
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umin.i64(i64 [[SHL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 2
  %max = call i64 @llvm.umin.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

; Multi-use Test Cases for `shl`
declare void @use(i64)

define i64 @test_shl_multi_use_add(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_multi_use_add(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    call void @use(i64 [[X1]])
; CHECK-NEXT:    [[TMP2:%.*]] = shl nuw i64 [[X]], 3
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[TMP2]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  call void @use(i64 %x1)
  %shl = shl nuw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

define i64 @test_shl_multi_use_shl(i64 %x) {
; CHECK-LABEL: define i64 @test_shl_multi_use_shl(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[SHL:%.*]] = shl nuw i64 [[X]], 2
; CHECK-NEXT:    call void @use(i64 [[SHL]])
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[SHL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %shl = shl nuw i64 %x, 2
  call void @use(i64 %shl)
  %max = call i64 @llvm.umax.i64(i64 %shl, i64 %x1)
  ret i64 %max
}

; Positive Test Cases for `mul`

define i64 @test_mul_by_3(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_by_3(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[MUL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_by_5(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_by_5(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 5
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[MUL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 5
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_with_nsw(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_with_nsw(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw nsw i64 [[X]], 3
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[MUL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw nsw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define <2 x i64> @test_mul_vector_by_3(<2 x i64> %x) {
; CHECK-LABEL: define <2 x i64> @test_mul_vector_by_3(
; CHECK-SAME: <2 x i64> [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw <2 x i64> [[X]], splat (i64 3)
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq <2 x i64> [[X]], zeroinitializer
; CHECK-NEXT:    [[MAX:%.*]] = select <2 x i1> [[TMP1]], <2 x i64> splat (i64 1), <2 x i64> [[MUL]]
; CHECK-NEXT:    ret <2 x i64> [[MAX]]
;
  %x1 = add <2 x i64> %x, <i64 1, i64 1>
  %mul = mul nuw <2 x i64> %x, <i64 3, i64 3>
  %max = call <2 x i64> @llvm.umax.v2i64(<2 x i64> %mul, <2 x i64> %x1)
  ret <2 x i64> %max
}

; Commuted Test Cases for `mul`

define i64 @test_mul_max_commuted(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_max_commuted(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[MUL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %x1, i64 %mul)
  ret i64 %max
}

; Negative Test Cases for `mul`

define i64 @test_mul_by_zero(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_by_zero(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    ret i64 [[X1]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 0
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_by_1(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_by_1(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[X]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 1
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_add_by_2(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_add_by_2(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 2
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[MUL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 2
  %mul = mul nuw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_without_nuw(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_without_nuw(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[MUL:%.*]] = mul i64 [[X]], 3
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[MUL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_umin(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_umin(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umin.i64(i64 [[MUL]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 3
  %max = call i64 @llvm.umin.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

; Multi-use Test Cases for `mul`

define i64 @test_mul_multi_use_add(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_multi_use_add(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[X1:%.*]] = add i64 [[X]], 1
; CHECK-NEXT:    call void @use(i64 [[X1]])
; CHECK-NEXT:    [[TMP2:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    [[MAX:%.*]] = call i64 @llvm.umax.i64(i64 [[TMP2]], i64 [[X1]])
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  call void @use(i64 %x1)
  %mul = mul nuw i64 %x, 3
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}

define i64 @test_mul_multi_use_mul(i64 %x) {
; CHECK-LABEL: define i64 @test_mul_multi_use_mul(
; CHECK-SAME: i64 [[X:%.*]]) {
; CHECK-NEXT:    [[MUL:%.*]] = mul nuw i64 [[X]], 3
; CHECK-NEXT:    call void @use(i64 [[MUL]])
; CHECK-NEXT:    [[TMP1:%.*]] = icmp eq i64 [[X]], 0
; CHECK-NEXT:    [[MAX:%.*]] = select i1 [[TMP1]], i64 1, i64 [[MUL]]
; CHECK-NEXT:    ret i64 [[MAX]]
;
  %x1 = add i64 %x, 1
  %mul = mul nuw i64 %x, 3
  call void @use(i64 %mul)
  %max = call i64 @llvm.umax.i64(i64 %mul, i64 %x1)
  ret i64 %max
}