Index: llvm/include/llvm/IR/IntrinsicsAArch64.td =================================================================== --- llvm/include/llvm/IR/IntrinsicsAArch64.td +++ llvm/include/llvm/IR/IntrinsicsAArch64.td @@ -959,6 +959,11 @@ llvm_i32_ty], [IntrNoMem]>; + class AdvSIMD_SVE_WHILE_Intrinsic + : Intrinsic<[llvm_anyvector_ty], + [llvm_anyint_ty, LLVMMatchType<1>], + [IntrNoMem]>; + class AdvSIMD_GatherLoad_64bitOffset_Intrinsic : Intrinsic<[llvm_anyvector_ty], [ @@ -1006,10 +1011,6 @@ // SVE let TargetPrefix = "aarch64" in { // All intrinsics start with "llvm.aarch64.". - class AdvSIMD_SVE_WHILE_Intrinsic - : Intrinsic<[llvm_anyvector_ty], - [llvm_anyint_ty, LLVMMatchType<1>], - [IntrNoMem]>; class AdvSIMD_GatherLoad_VecTorBase_Intrinsic : Intrinsic<[llvm_anyvector_ty], Index: llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td =================================================================== --- llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td +++ llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td @@ -1511,16 +1511,15 @@ defm TBX_ZZZ : sve2_int_perm_tbx<"tbx">; // SVE2 integer compare scalar count and limit - defm WHILEGE_PWW : sve_int_while4_rr<0b000, "whilege", null_frag>; - defm WHILEGT_PWW : sve_int_while4_rr<0b001, "whilegt", null_frag>; - defm WHILEHS_PWW : sve_int_while4_rr<0b100, "whilehs", null_frag>; - defm WHILEHI_PWW : sve_int_while4_rr<0b101, "whilehi", null_frag>; - - defm WHILEGE_PXX : sve_int_while8_rr<0b000, "whilege", null_frag>; - defm WHILEGT_PXX : sve_int_while8_rr<0b001, "whilegt", null_frag>; - defm WHILEHS_PXX : sve_int_while8_rr<0b100, "whilehs", null_frag>; - defm WHILEHI_PXX : sve_int_while8_rr<0b101, "whilehi", null_frag>; - + defm WHILEGE_PWW : sve_int_while4_rr<0b000, "whilege", int_aarch64_sve_whilege>; + defm WHILEGT_PWW : sve_int_while4_rr<0b001, "whilegt", int_aarch64_sve_whilegt>; + defm WHILEHS_PWW : sve_int_while4_rr<0b100, "whilehs", int_aarch64_sve_whilehs>; + defm WHILEHI_PWW : sve_int_while4_rr<0b101, "whilehi", int_aarch64_sve_whilehi>; + + defm WHILEGE_PXX : sve_int_while8_rr<0b000, "whilege", int_aarch64_sve_whilege>; + defm WHILEGT_PXX : sve_int_while8_rr<0b001, "whilegt", int_aarch64_sve_whilegt>; + defm WHILEHS_PXX : sve_int_while8_rr<0b100, "whilehs", int_aarch64_sve_whilehs>; + defm WHILEHI_PXX : sve_int_while8_rr<0b101, "whilehi", int_aarch64_sve_whilehi>; // SVE2 pointer conflict compare defm WHILEWR_PXX : sve2_int_while_rr<0b0, "whilewr">; Index: llvm/test/CodeGen/AArch64/sve2-intrinsics-while.ll =================================================================== --- /dev/null +++ llvm/test/CodeGen/AArch64/sve2-intrinsics-while.ll @@ -0,0 +1,309 @@ +; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve2 < %s | FileCheck %s + +; +; WHILEGE +; + +define @whilege_b_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilege_b_ww: +; CHECK: whilege p0.b, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv16i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilege_b_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilege_b_xx: +; CHECK: whilege p0.b, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv16i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilege_h_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilege_h_ww: +; CHECK: whilege p0.h, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv8i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilege_h_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilege_h_xx: +; CHECK: whilege p0.h, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv8i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilege_s_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilege_s_ww: +; CHECK: whilege p0.s, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv4i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilege_s_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilege_s_xx: +; CHECK: whilege p0.s, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv4i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilege_d_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilege_d_ww: +; CHECK: whilege p0.d, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv2i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilege_d_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilege_d_xx: +; CHECK: whilege p0.d, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilege.nxv2i1.i64(i64 %a, i64 %b) + ret %out +} + +; +; WHILEHS +; + +define @whilehs_b_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehs_b_ww: +; CHECK: whilehs p0.b, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv16i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehs_b_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehs_b_xx: +; CHECK: whilehs p0.b, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv16i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehs_h_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehs_h_ww: +; CHECK: whilehs p0.h, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv8i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehs_h_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehs_h_xx: +; CHECK: whilehs p0.h, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv8i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehs_s_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehs_s_ww: +; CHECK: whilehs p0.s, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv4i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehs_s_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehs_s_xx: +; CHECK: whilehs p0.s, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv4i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehs_d_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehs_d_ww: +; CHECK: whilehs p0.d, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv2i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehs_d_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehs_d_xx: +; CHECK: whilehs p0.d, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehs.nxv2i1.i64(i64 %a, i64 %b) + ret %out +} + +; +; WHILEGT +; + +define @whilegt_b_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilegt_b_ww: +; CHECK: whilegt p0.b, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv16i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilegt_b_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilegt_b_xx: +; CHECK: whilegt p0.b, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv16i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilegt_h_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilegt_h_ww: +; CHECK: whilegt p0.h, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv8i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilegt_h_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilegt_h_xx: +; CHECK: whilegt p0.h, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv8i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilegt_s_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilegt_s_ww: +; CHECK: whilegt p0.s, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv4i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilegt_s_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilegt_s_xx: +; CHECK: whilegt p0.s, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv4i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilegt_d_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilegt_d_ww: +; CHECK: whilegt p0.d, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv2i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilegt_d_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilegt_d_xx: +; CHECK: whilegt p0.d, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilegt.nxv2i1.i64(i64 %a, i64 %b) + ret %out +} + +; +; WHILEHI +; + +define @whilehi_b_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehi_b_ww: +; CHECK: whilehi p0.b, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv16i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehi_b_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehi_b_xx: +; CHECK: whilehi p0.b, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv16i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehi_h_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehi_h_ww: +; CHECK: whilehi p0.h, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv8i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehi_h_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehi_h_xx: +; CHECK: whilehi p0.h, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv8i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehi_s_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehi_s_ww: +; CHECK: whilehi p0.s, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv4i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehi_s_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehi_s_xx: +; CHECK: whilehi p0.s, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv4i1.i64(i64 %a, i64 %b) + ret %out +} + +define @whilehi_d_ww(i32 %a, i32 %b) { +; CHECK-LABEL: whilehi_d_ww: +; CHECK: whilehi p0.d, w0, w1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv2i1.i32(i32 %a, i32 %b) + ret %out +} + +define @whilehi_d_xx(i64 %a, i64 %b) { +; CHECK-LABEL: whilehi_d_xx: +; CHECK: whilehi p0.d, x0, x1 +; CHECK-NEXT: ret + %out = call @llvm.aarch64.sve.whilehi.nxv2i1.i64(i64 %a, i64 %b) + ret %out +} + +declare @llvm.aarch64.sve.whilege.nxv16i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilege.nxv16i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilege.nxv8i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilege.nxv8i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilege.nxv4i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilege.nxv4i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilege.nxv2i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilege.nxv2i1.i64(i64, i64) + +declare @llvm.aarch64.sve.whilehs.nxv16i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehs.nxv16i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehs.nxv8i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehs.nxv8i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehs.nxv4i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehs.nxv4i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehs.nxv2i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehs.nxv2i1.i64(i64, i64) + +declare @llvm.aarch64.sve.whilegt.nxv16i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilegt.nxv16i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilegt.nxv8i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilegt.nxv8i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilegt.nxv4i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilegt.nxv4i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilegt.nxv2i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilegt.nxv2i1.i64(i64, i64) + +declare @llvm.aarch64.sve.whilehi.nxv16i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehi.nxv16i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehi.nxv8i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehi.nxv8i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehi.nxv4i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehi.nxv4i1.i64(i64, i64) +declare @llvm.aarch64.sve.whilehi.nxv2i1.i32(i32, i32) +declare @llvm.aarch64.sve.whilehi.nxv2i1.i64(i64, i64)