diff --git a/llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td b/llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td --- a/llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td +++ b/llvm/lib/Target/AArch64/AArch64SVEInstrInfo.td @@ -375,6 +375,9 @@ (sub node:$op1, (AArch64mul_p_oneuse node:$pred, node:$op2, node:$op3)), // sub(a, select(mask, mul(b, c), splat(0))) -> mls(a, mask, b, c) (sub node:$op1, (vselect node:$pred, (AArch64mul_p_oneuse (SVEAllActive), node:$op2, node:$op3), (SVEDup0)))]>; +def AArch64eor3 : PatFrags<(ops node:$op1, node:$op2, node:$op3), + [(int_aarch64_sve_eor3 node:$op1, node:$op2, node:$op3), + (xor node:$op1, (xor node:$op2, node:$op3))]>; class fma_patfrags : PatFrags<(ops node:$pred, node:$op1, node:$op2, node:$op3), @@ -3527,7 +3530,7 @@ defm FMLSLT_ZZZ_SHH : sve2_fp_mla_long<0b11, "fmlslt", int_aarch64_sve_fmlslt>; // SVE2 bitwise ternary operations - defm EOR3_ZZZZ : sve2_int_bitwise_ternary_op<0b000, "eor3", int_aarch64_sve_eor3>; + defm EOR3_ZZZZ : sve2_int_bitwise_ternary_op<0b000, "eor3", AArch64eor3>; defm BCAX_ZZZZ : sve2_int_bitwise_ternary_op<0b010, "bcax", int_aarch64_sve_bcax>; defm BSL_ZZZZ : sve2_int_bitwise_ternary_op<0b001, "bsl", int_aarch64_sve_bsl, AArch64bsp>; defm BSL1N_ZZZZ : sve2_int_bitwise_ternary_op<0b011, "bsl1n", int_aarch64_sve_bsl1n>; diff --git a/llvm/test/CodeGen/AArch64/sve2-eor3.ll b/llvm/test/CodeGen/AArch64/sve2-eor3.ll new file mode 100644 --- /dev/null +++ b/llvm/test/CodeGen/AArch64/sve2-eor3.ll @@ -0,0 +1,151 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; RUN: llc -mtriple=aarch64-none-eabi -mattr=+sve < %s -o - | FileCheck --check-prefix=SVE %s +; RUN: llc -mtriple=aarch64-none-eabi -mattr=+sve2 < %s -o - | FileCheck --check-prefix=SVE2 %s + +define @eor3_nxv16i8_left( %0, %1, %2) { +; SVE-LABEL: eor3_nxv16i8_left: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z0.d, z2.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv16i8_left: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z0.d, z0.d, z1.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %4, %2 + ret %5 +} + +define @eor3_nxv16i8_right( %0, %1, %2) { +; SVE-LABEL: eor3_nxv16i8_right: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z2.d, z0.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv16i8_right: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z2.d, z2.d, z0.d, z1.d +; SVE2-NEXT: mov z0.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %2, %4 + ret %5 +} + +define @eor3_nxv8i16_left( %0, %1, %2) { +; SVE-LABEL: eor3_nxv8i16_left: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z0.d, z2.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv8i16_left: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z0.d, z0.d, z1.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %4, %2 + ret %5 +} + +define @eor3_nxv8i16_right( %0, %1, %2) { +; SVE-LABEL: eor3_nxv8i16_right: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z2.d, z0.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv8i16_right: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z2.d, z2.d, z0.d, z1.d +; SVE2-NEXT: mov z0.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %2, %4 + ret %5 +} + +define @eor3_nxv4i32_left( %0, %1, %2) { +; SVE-LABEL: eor3_nxv4i32_left: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z0.d, z2.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv4i32_left: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z0.d, z0.d, z1.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %4, %2 + ret %5 +} + +define @eor3_nxv4i32_right( %0, %1, %2) { +; SVE-LABEL: eor3_nxv4i32_right: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z2.d, z0.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv4i32_right: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z2.d, z2.d, z0.d, z1.d +; SVE2-NEXT: mov z0.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %2, %4 + ret %5 +} + +define @eor3_nxv2i64_left( %0, %1, %2) { +; SVE-LABEL: eor3_nxv2i64_left: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z0.d, z2.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv2i64_left: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z0.d, z0.d, z1.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %4, %2 + ret %5 +} + +define @eor3_nxv2i64_right( %0, %1, %2) { +; SVE-LABEL: eor3_nxv2i64_right: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: eor z0.d, z2.d, z0.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_nxv2i64_right: +; SVE2: // %bb.0: +; SVE2-NEXT: eor3 z2.d, z2.d, z0.d, z1.d +; SVE2-NEXT: mov z0.d, z2.d +; SVE2-NEXT: ret + %4 = xor %0, %1 + %5 = xor %2, %4 + ret %5 +} + +define @eor3_vnot( %0, %1) { +; SVE-LABEL: eor3_vnot: +; SVE: // %bb.0: +; SVE-NEXT: eor z0.d, z0.d, z1.d +; SVE-NEXT: ret +; +; SVE2-LABEL: eor3_vnot: +; SVE2: // %bb.0: +; SVE2-NEXT: eor z0.d, z0.d, z1.d +; SVE2-NEXT: ret + %3 = xor %0, zeroinitializer + %4 = xor %3, %1 + ret %4 +} +