llvm-project/llvm/test/CodeGen/AArch64/check-sign-bit-before-exten...

190 lines
4.6 KiB
LLVM

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc < %s -mtriple=aarch64-gnu-linux -o - | FileCheck %s
; These tests make sure that the `cmp` instruction is rendered with an
; instruction that checks the sign bit of the original unextended data
; (%in) instead of the sign bit of the sign extended one that is
; created by the type legalization process.
;
; The tests are subdivided in tests that determine the sign bit
; looking through a `sign_extend_inreg` and tests that determine the
; sign bit looking through a `sign_extend`.
define i32 @f_i8_sign_extend_inreg(i8 %in, i32 %a, i32 %b) nounwind {
; CHECK-LABEL: f_i8_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: sxtb w8, w0
; CHECK-NEXT: cmp w8, #0 // =0
; CHECK-NEXT: csel w8, w1, w2, ge
; CHECK-NEXT: add w0, w8, w0, uxtb
; CHECK-NEXT: ret
entry:
%cmp = icmp sgt i8 %in, -1
%ext = zext i8 %in to i32
br i1 %cmp, label %A, label %B
A:
%retA = add i32 %ext, %a
ret i32 %retA
B:
%retB = add i32 %ext, %b
ret i32 %retB
}
define i32 @f_i16_sign_extend_inreg(i16 %in, i32 %a, i32 %b) nounwind {
; CHECK-LABEL: f_i16_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: sxth w8, w0
; CHECK-NEXT: cmp w8, #0 // =0
; CHECK-NEXT: csel w8, w1, w2, ge
; CHECK-NEXT: add w0, w8, w0, uxth
; CHECK-NEXT: ret
entry:
%cmp = icmp sgt i16 %in, -1
%ext = zext i16 %in to i32
br i1 %cmp, label %A, label %B
A:
%retA = add i32 %ext, %a
ret i32 %retA
B:
%retB = add i32 %ext, %b
ret i32 %retB
}
define i64 @f_i32_sign_extend_inreg(i32 %in, i64 %a, i64 %b) nounwind {
; CHECK-LABEL: f_i32_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: cmp w0, #0 // =0
; CHECK-NEXT: csel x8, x1, x2, ge
; CHECK-NEXT: add x0, x8, w0, uxtw
; CHECK-NEXT: ret
entry:
%cmp = icmp sgt i32 %in, -1
%ext = zext i32 %in to i64
br i1 %cmp, label %A, label %B
A:
%retA = add i64 %ext, %a
ret i64 %retA
B:
%retB = add i64 %ext, %b
ret i64 %retB
}
define i32 @g_i8_sign_extend_inreg(i8 %in, i32 %a, i32 %b) nounwind {
; CHECK-LABEL: g_i8_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: sxtb w8, w0
; CHECK-NEXT: cmp w8, #0 // =0
; CHECK-NEXT: csel w8, w1, w2, lt
; CHECK-NEXT: add w0, w8, w0, uxtb
; CHECK-NEXT: ret
entry:
%cmp = icmp slt i8 %in, 0
%ext = zext i8 %in to i32
br i1 %cmp, label %A, label %B
A:
%retA = add i32 %ext, %a
ret i32 %retA
B:
%retB = add i32 %ext, %b
ret i32 %retB
}
define i32 @g_i16_sign_extend_inreg(i16 %in, i32 %a, i32 %b) nounwind {
; CHECK-LABEL: g_i16_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: sxth w8, w0
; CHECK-NEXT: cmp w8, #0 // =0
; CHECK-NEXT: csel w8, w1, w2, lt
; CHECK-NEXT: add w0, w8, w0, uxth
; CHECK-NEXT: ret
entry:
%cmp = icmp slt i16 %in, 0
%ext = zext i16 %in to i32
br i1 %cmp, label %A, label %B
A:
%retA = add i32 %ext, %a
ret i32 %retA
B:
%retB = add i32 %ext, %b
ret i32 %retB
}
define i64 @g_i32_sign_extend_inreg(i32 %in, i64 %a, i64 %b) nounwind {
; CHECK-LABEL: g_i32_sign_extend_inreg:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: cmp w0, #0 // =0
; CHECK-NEXT: csel x8, x1, x2, lt
; CHECK-NEXT: add x0, x8, w0, uxtw
; CHECK-NEXT: ret
entry:
%cmp = icmp slt i32 %in, 0
%ext = zext i32 %in to i64
br i1 %cmp, label %A, label %B
A:
%retA = add i64 %ext, %a
ret i64 %retA
B:
%retB = add i64 %ext, %b
ret i64 %retB
}
define i64 @f_i32_sign_extend_i64(i32 %in, i64 %a, i64 %b) nounwind {
; CHECK-LABEL: f_i32_sign_extend_i64:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
; CHECK-NEXT: sxtw x8, w0
; CHECK-NEXT: cmp x8, #0 // =0
; CHECK-NEXT: csel x8, x1, x2, ge
; CHECK-NEXT: add x0, x8, w0, uxtw
; CHECK-NEXT: ret
entry:
%inext = sext i32 %in to i64
%cmp = icmp sgt i64 %inext, -1
%ext = zext i32 %in to i64
br i1 %cmp, label %A, label %B
A:
%retA = add i64 %ext, %a
ret i64 %retA
B:
%retB = add i64 %ext, %b
ret i64 %retB
}
define i64 @g_i32_sign_extend_i64(i32 %in, i64 %a, i64 %b) nounwind {
; CHECK-LABEL: g_i32_sign_extend_i64:
; CHECK: // %bb.0: // %entry
; CHECK-NEXT: // kill: def $w0 killed $w0 def $x0
; CHECK-NEXT: sxtw x8, w0
; CHECK-NEXT: cmp x8, #0 // =0
; CHECK-NEXT: csel x8, x1, x2, lt
; CHECK-NEXT: add x0, x8, w0, uxtw
; CHECK-NEXT: ret
entry:
%inext = sext i32 %in to i64
%cmp = icmp slt i64 %inext, 0
%ext = zext i32 %in to i64
br i1 %cmp, label %A, label %B
A:
%retA = add i64 %ext, %a
ret i64 %retA
B:
%retB = add i64 %ext, %b
ret i64 %retB
}