[llvm] [AArch64][FEAT_CMPBR] Codegen for Armv9.6-a compare-and-branch (PR #116465)
via llvm-commits
llvm-commits at lists.llvm.org
Tue Feb 11 04:19:21 PST 2025
================
@@ -0,0 +1,1060 @@
+; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 5
+; RUN: llc -mtriple arm64-apple-ios -mattr +cmpbr -verify-machineinstrs -o - < %s | FileCheck %s --check-prefix=CHECK-CMPBR
+; RUN: llc -mtriple arm64-apple-ios -mattr -cmpbr -verify-machineinstrs -o - < %s | FileCheck %s --check-prefix=CHECK-NO-CMPBR
+
+; slt, sle, sgt, sge, ult, ule, ugt, uge, eq, ne
+
+define void @slt_0(i32 %a) {
+; CHECK-CMPBR-LABEL: slt_0:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: tbnz w0, #31, LBB0_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB0_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: slt_0:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: tbnz w0, #31, LBB0_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB0_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp slt i32 %a, 0
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @slt_m1(i32 %a) {
+; CHECK-CMPBR-LABEL: slt_m1:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #-2 ; =0xfffffffe
+; CHECK-CMPBR-NEXT: cbge w8, w0, LBB1_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB1_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: slt_m1:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmn w0, #2
+; CHECK-NO-CMPBR-NEXT: b.le LBB1_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB1_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp slt i32 %a, -1
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @slt_63(i32 %a) {
+; CHECK-CMPBR-LABEL: slt_63:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: cblt w0, #63, LBB2_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB2_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: slt_63:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #62
+; CHECK-NO-CMPBR-NEXT: b.le LBB2_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB2_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp slt i32 %a, 63
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @slt_64(i32 %a) {
+; CHECK-CMPBR-LABEL: slt_64:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #63 ; =0x3f
+; CHECK-CMPBR-NEXT: cbge w8, w0, LBB3_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB3_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: slt_64:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #63
+; CHECK-NO-CMPBR-NEXT: b.le LBB3_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB3_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp slt i32 %a, 64
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sle_0(i32 %a) {
+; CHECK-CMPBR-LABEL: sle_0:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: cblt w0, #1, LBB4_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB4_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sle_0:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #0
+; CHECK-NO-CMPBR-NEXT: b.le LBB4_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB4_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sle i32 %a, 0
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sle_m1(i32 %a) {
+; CHECK-CMPBR-LABEL: sle_m1:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: tbnz w0, #31, LBB5_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB5_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sle_m1:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: tbnz w0, #31, LBB5_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB5_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sle i32 %a, -1
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sle_63(i32 %a) {
+; CHECK-CMPBR-LABEL: sle_63:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #63 ; =0x3f
+; CHECK-CMPBR-NEXT: cbge w8, w0, LBB6_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB6_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sle_63:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #63
+; CHECK-NO-CMPBR-NEXT: b.le LBB6_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB6_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sle i32 %a, 63
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sle_64(i32 %a) {
+; CHECK-CMPBR-LABEL: sle_64:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #64 ; =0x40
+; CHECK-CMPBR-NEXT: cbge w8, w0, LBB7_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB7_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sle_64:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #64
+; CHECK-NO-CMPBR-NEXT: b.le LBB7_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB7_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sle i32 %a, 64
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sgt_0(i32 %a) {
+; CHECK-CMPBR-LABEL: sgt_0:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: cbgt w0, #0, LBB8_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB8_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sgt_0:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #1
+; CHECK-NO-CMPBR-NEXT: b.ge LBB8_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB8_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sgt i32 %a, 0
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sgt_m1(i32 %a) {
+; CHECK-CMPBR-LABEL: sgt_m1:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: tbz w0, #31, LBB9_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB9_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sgt_m1:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: tbz w0, #31, LBB9_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB9_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sgt i32 %a, -1
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sgt_63(i32 %a) {
+; CHECK-CMPBR-LABEL: sgt_63:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #64 ; =0x40
+; CHECK-CMPBR-NEXT: cbge w0, w8, LBB10_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB10_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sgt_63:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #64
+; CHECK-NO-CMPBR-NEXT: b.ge LBB10_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB10_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sgt i32 %a, 63
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sgt_64(i32 %a) {
+; CHECK-CMPBR-LABEL: sgt_64:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #65 ; =0x41
+; CHECK-CMPBR-NEXT: cbge w0, w8, LBB11_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB11_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sgt_64:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #65
+; CHECK-NO-CMPBR-NEXT: b.ge LBB11_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB11_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sgt i32 %a, 64
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sge_0(i32 %a) {
+; CHECK-CMPBR-LABEL: sge_0:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: tbz w0, #31, LBB12_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB12_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sge_0:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: tbz w0, #31, LBB12_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB12_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sge i32 %a, 0
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sge_m1(i32 %a) {
+; CHECK-CMPBR-LABEL: sge_m1:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #-1 ; =0xffffffff
+; CHECK-CMPBR-NEXT: cbge w0, w8, LBB13_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB13_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sge_m1:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmn w0, #1
+; CHECK-NO-CMPBR-NEXT: b.ge LBB13_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB13_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sge i32 %a, -1
+ br i1 %cmp, label %if.then, label %if.end
+
+if.then:
+ tail call void @llvm.trap()
+ unreachable
+
+if.end:
+ ret void
+}
+
+define void @sge_63(i32 %a) {
+; CHECK-CMPBR-LABEL: sge_63:
+; CHECK-CMPBR: ; %bb.0: ; %entry
+; CHECK-CMPBR-NEXT: mov w8, #64 ; =0x40
+; CHECK-CMPBR-NEXT: cbge w0, w8, LBB14_2
+; CHECK-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-CMPBR-NEXT: ret
+; CHECK-CMPBR-NEXT: LBB14_2: ; %if.then
+; CHECK-CMPBR-NEXT: brk #0x1
+;
+; CHECK-NO-CMPBR-LABEL: sge_63:
+; CHECK-NO-CMPBR: ; %bb.0: ; %entry
+; CHECK-NO-CMPBR-NEXT: cmp w0, #64
+; CHECK-NO-CMPBR-NEXT: b.ge LBB14_2
+; CHECK-NO-CMPBR-NEXT: ; %bb.1: ; %if.end
+; CHECK-NO-CMPBR-NEXT: ret
+; CHECK-NO-CMPBR-NEXT: LBB14_2: ; %if.then
+; CHECK-NO-CMPBR-NEXT: brk #0x1
+entry:
+ %cmp = icmp sgt i32 %a, 63
----------------
SpencerAbson wrote:
Should this (and the test below) use `sge`?
https://github.com/llvm/llvm-project/pull/116465
More information about the llvm-commits
mailing list