; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve -mattr=+use-experimental-zeroing-pseudos < %s | FileCheck %s ; ; ADD ; define @add_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: add_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: add z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.add.nxv16i8( %pg, %a_z, %b) ret %out } define @add_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: add_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: add z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.add.nxv8i16( %pg, %a_z, %b) ret %out } define @add_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: add_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: add z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.add.nxv4i32( %pg, %a_z, %b) ret %out } define @add_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: add_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: add z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.add.nxv2i64( %pg, %a_z, %b) ret %out } ; ; SUB ; define @sub_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: sub_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: sub z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.sub.nxv16i8( %pg, %a_z, %b) ret %out } define @sub_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: sub_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: sub z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.sub.nxv8i16( %pg, %a_z, %b) ret %out } define @sub_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: sub_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: sub z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.sub.nxv4i32( %pg, %a_z, %b) ret %out } define @sub_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: sub_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: sub z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.sub.nxv2i64( %pg, %a_z, %b) ret %out } ; ; SUBR ; define @subr_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: subr_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: subr z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.subr.nxv16i8( %pg, %a_z, %b) ret %out } define @subr_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: subr_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: subr z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.subr.nxv8i16( %pg, %a_z, %b) ret %out } define @subr_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: subr_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: subr z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.subr.nxv4i32( %pg, %a_z, %b) ret %out } define @subr_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: subr_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: subr z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.subr.nxv2i64( %pg, %a_z, %b) ret %out } ; ; ORR ; define @orr_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: orr_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: orr z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.orr.nxv16i8( %pg, %a_z, %b) ret %out } define @orr_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: orr_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: orr z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.orr.nxv8i16( %pg, %a_z, %b) ret %out } define @orr_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: orr_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: orr z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.orr.nxv4i32( %pg, %a_z, %b) ret %out } define @orr_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: orr_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: orr z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.orr.nxv2i64( %pg, %a_z, %b) ret %out } ; ; EOR ; define @eor_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: eor_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: eor z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.eor.nxv16i8( %pg, %a_z, %b) ret %out } define @eor_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: eor_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: eor z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.eor.nxv8i16( %pg, %a_z, %b) ret %out } define @eor_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: eor_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: eor z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.eor.nxv4i32( %pg, %a_z, %b) ret %out } define @eor_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: eor_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: eor z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.eor.nxv2i64( %pg, %a_z, %b) ret %out } ; ; AND ; define @and_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: and_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: and z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.and.nxv16i8( %pg, %a_z, %b) ret %out } define @and_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: and_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: and z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.and.nxv8i16( %pg, %a_z, %b) ret %out } define @and_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: and_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: and z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.and.nxv4i32( %pg, %a_z, %b) ret %out } define @and_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: and_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: and z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.and.nxv2i64( %pg, %a_z, %b) ret %out } ; ; BIC ; define @bic_i8_zero( %pg, %a, %b) { ; CHECK-LABEL: bic_i8_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.b, p0/z, z0.b ; CHECK-NEXT: bic z0.b, p0/m, z0.b, z1.b ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv16i8( %pg, %a_z, %b) ret %out } define @bic_i16_zero( %pg, %a, %b) { ; CHECK-LABEL: bic_i16_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.h, p0/z, z0.h ; CHECK-NEXT: bic z0.h, p0/m, z0.h, z1.h ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv8i16( %pg, %a_z, %b) ret %out } define @bic_i32_zero( %pg, %a, %b) { ; CHECK-LABEL: bic_i32_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.s, p0/z, z0.s ; CHECK-NEXT: bic z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv4i32( %pg, %a_z, %b) ret %out } define @bic_i64_zero( %pg, %a, %b) { ; CHECK-LABEL: bic_i64_zero: ; CHECK: // %bb.0: ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: bic z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv2i64( %pg, %a_z, %b) ret %out } ; BIC (i.e. A & ~A) is illegal operation with movprfx, so the codegen depend on IR before expand-pseudo define @bic_i64_zero_no_unique_reg( %pg, %a) { ; CHECK-LABEL: bic_i64_zero_no_unique_reg: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z1.d, #0 // =0x0 ; CHECK-NEXT: mov z1.d, p0/m, z0.d ; CHECK-NEXT: movprfx z0.d, p0/z, z0.d ; CHECK-NEXT: bic z0.d, p0/m, z0.d, z1.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv2i64( %pg, %a_z, %a_z) ret %out } ; BIC (i.e. A & ~B) is not a commutative operation, so disable it when the ; destination operand is not the destructive operand define @bic_i64_zero_no_comm( %pg, %a, %b) { ; CHECK-LABEL: bic_i64_zero_no_comm: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z2.d, #0 // =0x0 ; CHECK-NEXT: mov z2.d, p0/m, z0.d ; CHECK-NEXT: mov z0.d, z1.d ; CHECK-NEXT: bic z0.d, p0/m, z0.d, z2.d ; CHECK-NEXT: ret %a_z = select %pg, %a, zeroinitializer %out = call @llvm.aarch64.sve.bic.nxv2i64( %pg, %b, %a_z) ret %out } declare @llvm.aarch64.sve.add.nxv16i8(, , ) declare @llvm.aarch64.sve.add.nxv8i16(, , ) declare @llvm.aarch64.sve.add.nxv4i32(, , ) declare @llvm.aarch64.sve.add.nxv2i64(, , ) declare @llvm.aarch64.sve.sub.nxv16i8(, , ) declare @llvm.aarch64.sve.sub.nxv8i16(, , ) declare @llvm.aarch64.sve.sub.nxv4i32(, , ) declare @llvm.aarch64.sve.sub.nxv2i64(, , ) declare @llvm.aarch64.sve.subr.nxv16i8(, , ) declare @llvm.aarch64.sve.subr.nxv8i16(, , ) declare @llvm.aarch64.sve.subr.nxv4i32(, , ) declare @llvm.aarch64.sve.subr.nxv2i64(, , ) declare @llvm.aarch64.sve.orr.nxv16i8(, , ) declare @llvm.aarch64.sve.orr.nxv8i16(, , ) declare @llvm.aarch64.sve.orr.nxv4i32(, , ) declare @llvm.aarch64.sve.orr.nxv2i64(, , ) declare @llvm.aarch64.sve.eor.nxv16i8(, , ) declare @llvm.aarch64.sve.eor.nxv8i16(, , ) declare @llvm.aarch64.sve.eor.nxv4i32(, , ) declare @llvm.aarch64.sve.eor.nxv2i64(, , ) declare @llvm.aarch64.sve.and.nxv16i8(, , ) declare @llvm.aarch64.sve.and.nxv8i16(, , ) declare @llvm.aarch64.sve.and.nxv4i32(, , ) declare @llvm.aarch64.sve.and.nxv2i64(, , ) declare @llvm.aarch64.sve.bic.nxv16i8(, , ) declare @llvm.aarch64.sve.bic.nxv8i16(, , ) declare @llvm.aarch64.sve.bic.nxv4i32(, , ) declare @llvm.aarch64.sve.bic.nxv2i64(, , )