llvm-project/llvm/test/CodeGen/RISCV/rv64xandesperf.ll
Craig Topper 2817873082
[RISCV] Fold (sext_inreg (setcc), i1) -> (sub 0, (setcc). (#154206)
This helps the 3 vendor extensions that make sext_inreg i1 legal.

I'm delaying this until after LegalizeDAG since we normally have
sext_inreg i1 up until LegalizeDAG turns it into and+neg.

I also delayed the recently added (sext_inreg (xor (setcc), -1), i1)
combine. Though the xor isn't likely to appear before LegalizeDAG anyway.
2025-08-18 21:24:03 -07:00

468 lines
11 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc -mtriple=riscv64 -mattr=+xandesperf -verify-machineinstrs < %s \
; RUN: | FileCheck %s
; NDS.BFOZ
; MSB >= LSB
define i32 @bfoz_from_and_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_and_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 11, 0
; CHECK-NEXT: ret
%a = and i32 %x, 4095
ret i32 %a
}
define i64 @bfoz_from_and_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_and_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 11, 0
; CHECK-NEXT: ret
%a = and i64 %x, 4095
ret i64 %a
}
define i32 @bfoz_from_and_lshr_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_and_lshr_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 25, 23
; CHECK-NEXT: ret
%shifted = lshr i32 %x, 23
%masked = and i32 %shifted, 7
ret i32 %masked
}
define i64 @bfoz_from_and_lshr_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_and_lshr_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 57, 46
; CHECK-NEXT: ret
%shifted = lshr i64 %x, 46
%masked = and i64 %shifted, 4095
ret i64 %masked
}
define i32 @bfoz_from_lshr_and_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_lshr_and_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 23, 12
; CHECK-NEXT: ret
%masked = and i32 %x, 16773120
%shifted = lshr i32 %masked, 12
ret i32 %shifted
}
define i64 @bfoz_from_lshr_and_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_lshr_and_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 35, 24
; CHECK-NEXT: ret
%masked = and i64 %x, 68702699520
%shifted = lshr i64 %masked, 24
ret i64 %shifted
}
; MSB = 0
define i32 @bfoz_from_and_shl_with_msb_zero_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_and_shl_with_msb_zero_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 0, 15
; CHECK-NEXT: ret
%shifted = shl i32 %x, 15
%masked = and i32 %shifted, 32768
ret i32 %masked
}
define i64 @bfoz_from_and_shl_with_msb_zero_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_and_shl_with_msb_zero_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 0, 48
; CHECK-NEXT: ret
%shifted = shl i64 %x, 48
%masked = and i64 %shifted, 281474976710656
ret i64 %masked
}
define i32 @bfoz_from_lshr_shl_with_msb_zero_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_lshr_shl_with_msb_zero_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 0, 18
; CHECK-NEXT: ret
%shl = shl i32 %x, 31
%lshr = lshr i32 %shl, 13
ret i32 %lshr
}
define i64 @bfoz_from_lshr_shl_with_msb_zero_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_lshr_shl_with_msb_zero_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 0, 44
; CHECK-NEXT: ret
%shl = shl i64 %x, 63
%lshr = lshr i64 %shl, 19
ret i64 %lshr
}
; MSB < LSB
define i32 @bfoz_from_and_shl_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_and_shl_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 12, 23
; CHECK-NEXT: ret
%shifted = shl i32 %x, 12
%masked = and i32 %shifted, 16773120
ret i32 %masked
}
define i64 @bfoz_from_and_shl_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_and_shl_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 24, 35
; CHECK-NEXT: ret
%shifted = shl i64 %x, 24
%masked = and i64 %shifted, 68702699520
ret i64 %masked
}
define i32 @bfoz_from_lshr_shl_i32(i32 %x) {
; CHECK-LABEL: bfoz_from_lshr_shl_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 19, 24
; CHECK-NEXT: ret
%shl = shl i32 %x, 26
%lshr = lshr i32 %shl, 7
ret i32 %lshr
}
define i64 @bfoz_from_lshr_shl_i64(i64 %x) {
; CHECK-LABEL: bfoz_from_lshr_shl_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfoz a0, a0, 25, 48
; CHECK-NEXT: ret
%shl = shl i64 %x, 40
%lshr = lshr i64 %shl, 15
ret i64 %lshr
}
; NDS.BFOS
; MSB >= LSB
define i32 @bfos_from_ashr_shl_i32(i32 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 23, 16
; CHECK-NEXT: ret
%shl = shl i32 %x, 8
%ashr = ashr i32 %shl, 24
ret i32 %ashr
}
define i64 @bfos_from_ashr_shl_i64(i64 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 55, 16
; CHECK-NEXT: ret
%shl = shl i64 %x, 8
%ashr = ashr i64 %shl, 24
ret i64 %ashr
}
define i32 @bfos_from_ashr_sexti8_i32(i8 %x) {
; CHECK-LABEL: bfos_from_ashr_sexti8_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 5
; CHECK-NEXT: ret
%sext = sext i8 %x to i32
%ashr = ashr i32 %sext, 5
ret i32 %ashr
}
define i64 @bfos_from_ashr_sexti8_i64(i8 %x) {
; CHECK-LABEL: bfos_from_ashr_sexti8_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 5
; CHECK-NEXT: ret
%sext = sext i8 %x to i64
%ashr = ashr i64 %sext, 5
ret i64 %ashr
}
define i32 @bfos_from_ashr_sexti16_i32(i16 %x) {
; CHECK-LABEL: bfos_from_ashr_sexti16_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 11
; CHECK-NEXT: ret
%sext = sext i16 %x to i32
%ashr = ashr i32 %sext, 11
ret i32 %ashr
}
define i64 @bfos_from_ashr_sexti16_i64(i16 %x) {
; CHECK-LABEL: bfos_from_ashr_sexti16_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 11
; CHECK-NEXT: ret
%sext = sext i16 %x to i64
%ashr = ashr i64 %sext, 11
ret i64 %ashr
}
; MSB = 0
define i32 @bfos_from_ashr_shl_with_msb_zero_insert_i32(i32 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_with_msb_zero_insert_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 14
; CHECK-NEXT: ret
%shl = shl i32 %x, 31
%lshr = ashr i32 %shl, 17
ret i32 %lshr
}
define i64 @bfos_from_ashr_shl_with_msb_zero_insert_i64(i64 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_with_msb_zero_insert_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 46
; CHECK-NEXT: ret
%shl = shl i64 %x, 63
%lshr = ashr i64 %shl, 17
ret i64 %lshr
}
; MSB < LSB
define i32 @bfos_from_ashr_shl_insert_i32(i32 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_insert_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 18, 20
; CHECK-NEXT: ret
%shl = shl i32 %x, 29
%lshr = ashr i32 %shl, 11
ret i32 %lshr
}
define i64 @bfos_from_ashr_shl_insert_i64(i64 %x) {
; CHECK-LABEL: bfos_from_ashr_shl_insert_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 18, 52
; CHECK-NEXT: ret
%shl = shl i64 %x, 29
%lshr = ashr i64 %shl, 11
ret i64 %lshr
}
; sext
define signext i32 @sexti1_i32(i32 signext %a) {
; CHECK-LABEL: sexti1_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 0
; CHECK-NEXT: ret
%shl = shl i32 %a, 31
%shr = ashr exact i32 %shl, 31
ret i32 %shr
}
define signext i32 @sexti1_i32_2(i1 %a) {
; CHECK-LABEL: sexti1_i32_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 0
; CHECK-NEXT: ret
%1 = sext i1 %a to i32
ret i32 %1
}
; Make sure we don't use not+nds.bfos
define zeroext i8 @sexti1_i32_setcc(i32 signext %a) {
; CHECK-LABEL: sexti1_i32_setcc:
; CHECK: # %bb.0:
; CHECK-NEXT: srli a0, a0, 63
; CHECK-NEXT: addi a0, a0, -1
; CHECK-NEXT: zext.b a0, a0
; CHECK-NEXT: ret
%icmp = icmp sgt i32 %a, -1
%sext = sext i1 %icmp to i8
ret i8 %sext
}
; Make sure we don't use seqz+nds.bfos instead of snez+addi
define signext i32 @sexti1_i32_setcc_2(i32 signext %a, i32 signext %b) {
; CHECK-LABEL: sexti1_i32_setcc_2:
; CHECK: # %bb.0:
; CHECK-NEXT: xor a0, a0, a1
; CHECK-NEXT: snez a0, a0
; CHECK-NEXT: addi a0, a0, -1
; CHECK-NEXT: ret
%icmp = icmp eq i32 %a, %b
%sext = sext i1 %icmp to i32
ret i32 %sext
}
; Make sure we don't use nds.bfos instead of neg.
define signext i32 @sexti1_i32_setcc_3(i32 signext %a, i32 signext %b) {
; CHECK-LABEL: sexti1_i32_setcc_3:
; CHECK: # %bb.0:
; CHECK-NEXT: slt a0, a0, a1
; CHECK-NEXT: neg a0, a0
; CHECK-NEXT: ret
%icmp = icmp slt i32 %a, %b
%sext = sext i1 %icmp to i32
ret i32 %sext
}
define signext i32 @sexti8_i32(i32 signext %a) {
; CHECK-LABEL: sexti8_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 0
; CHECK-NEXT: ret
%shl = shl i32 %a, 24
%shr = ashr exact i32 %shl, 24
ret i32 %shr
}
define signext i32 @sexti8_i32_2(i8 %a) {
; CHECK-LABEL: sexti8_i32_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 0
; CHECK-NEXT: ret
%1 = sext i8 %a to i32
ret i32 %1
}
define signext i32 @sexti16_i32(i32 signext %a) {
; CHECK-LABEL: sexti16_i32:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 0
; CHECK-NEXT: ret
%shl = shl i32 %a, 16
%shr = ashr exact i32 %shl, 16
ret i32 %shr
}
define signext i32 @sexti16_i32_2(i16 %a) {
; CHECK-LABEL: sexti16_i32_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 0
; CHECK-NEXT: ret
%1 = sext i16 %a to i32
ret i32 %1
}
define i64 @sexti1_i64(i64 %a) {
; CHECK-LABEL: sexti1_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 0
; CHECK-NEXT: ret
%shl = shl i64 %a, 63
%shr = ashr exact i64 %shl, 63
ret i64 %shr
}
define i64 @sexti1_i64_2(i1 %a) {
; CHECK-LABEL: sexti1_i64_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 0, 0
; CHECK-NEXT: ret
%1 = sext i1 %a to i64
ret i64 %1
}
; Make sure we don't use not+nds.bfos
define zeroext i8 @sexti1_i64_setcc(i64 %a) {
; CHECK-LABEL: sexti1_i64_setcc:
; CHECK: # %bb.0:
; CHECK-NEXT: srli a0, a0, 63
; CHECK-NEXT: addi a0, a0, -1
; CHECK-NEXT: zext.b a0, a0
; CHECK-NEXT: ret
%icmp = icmp sgt i64 %a, -1
%sext = sext i1 %icmp to i8
ret i8 %sext
}
; Make sure we don't use seqz+nds.bfos instead of snez+addi
define i64 @sexti1_i64_setcc_2(i64 %a, i64 %b) {
; CHECK-LABEL: sexti1_i64_setcc_2:
; CHECK: # %bb.0:
; CHECK-NEXT: xor a0, a0, a1
; CHECK-NEXT: snez a0, a0
; CHECK-NEXT: addi a0, a0, -1
; CHECK-NEXT: ret
%icmp = icmp eq i64 %a, %b
%sext = sext i1 %icmp to i64
ret i64 %sext
}
; Make sure we don't use nds.bfos instead of neg.
define i64 @sexti1_i64_setcc_3(i64 %a, i64 %b) {
; CHECK-LABEL: sexti1_i64_setcc_3:
; CHECK: # %bb.0:
; CHECK-NEXT: slt a0, a0, a1
; CHECK-NEXT: neg a0, a0
; CHECK-NEXT: ret
%icmp = icmp slt i64 %a, %b
%sext = sext i1 %icmp to i64
ret i64 %sext
}
define i64 @sexti8_i64(i64 %a) {
; CHECK-LABEL: sexti8_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 0
; CHECK-NEXT: ret
%shl = shl i64 %a, 56
%shr = ashr exact i64 %shl, 56
ret i64 %shr
}
define i64 @sexti8_i64_2(i8 %a) {
; CHECK-LABEL: sexti8_i64_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 7, 0
; CHECK-NEXT: ret
%1 = sext i8 %a to i64
ret i64 %1
}
define i64 @sexti16_i64(i64 %a) {
; CHECK-LABEL: sexti16_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 0
; CHECK-NEXT: ret
%shl = shl i64 %a, 48
%shr = ashr exact i64 %shl, 48
ret i64 %shr
}
define i64 @sexti16_i64_2(i16 %a) {
; CHECK-LABEL: sexti16_i64_2:
; CHECK: # %bb.0:
; CHECK-NEXT: nds.bfos a0, a0, 15, 0
; CHECK-NEXT: ret
%1 = sext i16 %a to i64
ret i64 %1
}
define i64 @sexti32_i64(i64 %a) {
; CHECK-LABEL: sexti32_i64:
; CHECK: # %bb.0:
; CHECK-NEXT: sext.w a0, a0
; CHECK-NEXT: ret
%shl = shl i64 %a, 32
%shr = ashr exact i64 %shl, 32
ret i64 %shr
}
define i64 @sexti32_i64_2(i32 signext %a) {
; CHECK-LABEL: sexti32_i64_2:
; CHECK: # %bb.0:
; CHECK-NEXT: ret
%1 = sext i32 %a to i64
ret i64 %1
}