; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve2 < %s | FileCheck %s ; ; ADD ; define @add_i8( %a, %b) { ; CHECK-LABEL: add_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: add z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.add.u.nxv16i8( %pg, %a, %b) ret %out } define @add_i16( %a, %b) { ; CHECK-LABEL: add_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: add z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.add.u.nxv8i16( %pg, %a, %b) ret %out } define @add_i32( %a, %b) { ; CHECK-LABEL: add_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: add z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.add.u.nxv4i32( %pg, %a, %b) ret %out } define @add_i64( %a, %b) { ; CHECK-LABEL: add_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: add z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.add.u.nxv2i64( %pg, %a, %b) ret %out } ; ; SUB ; define @sub_i8( %a, %b) { ; CHECK-LABEL: sub_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.sub.u.nxv16i8( %pg, %a, %b) ret %out } define @sub_i16( %a, %b) { ; CHECK-LABEL: sub_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.sub.u.nxv8i16( %pg, %a, %b) ret %out } define @sub_i32( %a, %b) { ; CHECK-LABEL: sub_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.sub.u.nxv4i32( %pg, %a, %b) ret %out } define @sub_i64( %a, %b) { ; CHECK-LABEL: sub_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.sub.u.nxv2i64( %pg, %a, %b) ret %out } ; As sub_i32 but where pg is i8 based and thus compatible for i32. define @sub_i32_ptrue_all_b( %a, %b) { ; CHECK-LABEL: sub_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.b = tail call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.sub.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As sub_i32 but where pg is i16 based and thus compatible for i32. define @sub_i32_ptrue_all_h( %a, %b) { ; CHECK-LABEL: sub_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: sub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.h = tail call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv8i1( %pg.h) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.sub.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As sub_i32 but where pg is i64 based, which is not compatibile for i32 and ; thus inactive lanes are important and the immediate form cannot be used. define @sub_i32_ptrue_all_d( %a, %b) { ; CHECK-LABEL: sub_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: sub z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %pg.d = tail call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv2i1( %pg.d) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.sub.nxv4i32( %pg.s, %a, %b) ret %out } ; ; MUL ; define @mul_i8( %a, %b) { ; CHECK-LABEL: mul_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: mul z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.mul.u.nxv16i8( %pg, %a, %b) ret %out } define @mul_i16( %a, %b) { ; CHECK-LABEL: mul_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: mul z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.mul.u.nxv8i16( %pg, %a, %b) ret %out } define @mul_i32( %a, %b) { ; CHECK-LABEL: mul_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: mul z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.mul.u.nxv4i32( %pg, %a, %b) ret %out } define @mul_i64( %a, %b) { ; CHECK-LABEL: mul_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: mul z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.mul.u.nxv2i64( %pg, %a, %b) ret %out } ; ; SMULH ; define @smulh_i8( %a, %b) { ; CHECK-LABEL: smulh_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: smulh z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.smulh.u.nxv16i8( %pg, %a, %b) ret %out } define @smulh_i16( %a, %b) { ; CHECK-LABEL: smulh_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: smulh z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.smulh.u.nxv8i16( %pg, %a, %b) ret %out } define @smulh_i32( %a, %b) { ; CHECK-LABEL: smulh_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: smulh z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.smulh.u.nxv4i32( %pg, %a, %b) ret %out } define @smulh_i64( %a, %b) { ; CHECK-LABEL: smulh_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: smulh z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.smulh.u.nxv2i64( %pg, %a, %b) ret %out } ; ; UMULH ; define @umulh_i8( %a, %b) { ; CHECK-LABEL: umulh_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.umulh.u.nxv16i8( %pg, %a, %b) ret %out } define @umulh_i16( %a, %b) { ; CHECK-LABEL: umulh_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.umulh.u.nxv8i16( %pg, %a, %b) ret %out } define @umulh_i32( %a, %b) { ; CHECK-LABEL: umulh_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.umulh.u.nxv4i32( %pg, %a, %b) ret %out } define @umulh_i64( %a, %b) { ; CHECK-LABEL: umulh_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.umulh.u.nxv2i64( %pg, %a, %b) ret %out } ; As umulh_i32 but where pg is i8 based and thus compatible for i32. define @umulh_i32_ptrue_all_b( %a, %b) { ; CHECK-LABEL: umulh_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.b = tail call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.umulh.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As umulh_i32 but where pg is i16 based and thus compatible for i32. define @umulh_i32_ptrue_all_h( %a, %b) { ; CHECK-LABEL: umulh_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: umulh z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.h = tail call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv8i1( %pg.h) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.umulh.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As umulh_i32 but where pg is i64 based, which is not compatibile for i32 and ; thus inactive lanes are important and the immediate form cannot be used. define @umulh_i32_ptrue_all_d( %a, %b) { ; CHECK-LABEL: umulh_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: umulh z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %pg.d = tail call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv2i1( %pg.d) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.umulh.nxv4i32( %pg.s, %a, %b) ret %out } ; ; AND ; define @and_i8( %a, %b) { ; CHECK-LABEL: and_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: and z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.and.u.nxv16i8( %pg, %a, %b) ret %out } define @and_i16( %a, %b) { ; CHECK-LABEL: and_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: and z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.and.u.nxv8i16( %pg, %a, %b) ret %out } define @and_i32( %a, %b) { ; CHECK-LABEL: and_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: and z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.and.u.nxv4i32( %pg, %a, %b) ret %out } define @and_i64( %a, %b) { ; CHECK-LABEL: and_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: and z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.and.u.nxv2i64( %pg, %a, %b) ret %out } ; ; BIC ; define @bic_i8( %a, %b) { ; CHECK-LABEL: bic_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: bic z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.bic.u.nxv16i8( %pg, %a, %b) ret %out } define @bic_i16( %a, %b) { ; CHECK-LABEL: bic_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: bic z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.bic.u.nxv8i16( %pg, %a, %b) ret %out } define @bic_i32( %a, %b) { ; CHECK-LABEL: bic_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: bic z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.bic.u.nxv4i32( %pg, %a, %b) ret %out } define @bic_i64( %a, %b) { ; CHECK-LABEL: bic_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: bic z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.bic.u.nxv2i64( %pg, %a, %b) ret %out } ; ; EOR ; define @eor_i8( %a, %b) { ; CHECK-LABEL: eor_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: eor z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.eor.u.nxv16i8( %pg, %a, %b) ret %out } define @eor_i16( %a, %b) { ; CHECK-LABEL: eor_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: eor z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.eor.u.nxv8i16( %pg, %a, %b) ret %out } define @eor_i32( %a, %b) { ; CHECK-LABEL: eor_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: eor z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.eor.u.nxv4i32( %pg, %a, %b) ret %out } define @eor_i64( %a, %b) { ; CHECK-LABEL: eor_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: eor z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.eor.u.nxv2i64( %pg, %a, %b) ret %out } ; ; ORR ; define @orr_i8( %a, %b) { ; CHECK-LABEL: orr_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.orr.u.nxv16i8( %pg, %a, %b) ret %out } define @orr_i16( %a, %b) { ; CHECK-LABEL: orr_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.orr.u.nxv8i16( %pg, %a, %b) ret %out } define @orr_i32( %a, %b) { ; CHECK-LABEL: orr_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.orr.u.nxv4i32( %pg, %a, %b) ret %out } define @orr_i64( %a, %b) { ; CHECK-LABEL: orr_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.orr.u.nxv2i64( %pg, %a, %b) ret %out } ; As orr_i32 but where pg is i8 based and thus compatible for i32. define @orr_i32_ptrue_all_b( %a, %b) { ; CHECK-LABEL: orr_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg.b = tail call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.orr.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As orr_i32 but where pg is i16 based and thus compatible for i32. define @orr_i32_ptrue_all_h( %a, %b) { ; CHECK-LABEL: orr_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: orr z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg.h = tail call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv8i1( %pg.h) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.orr.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As orr_i32 but where pg is i64 based, which is not compatibile for i32 and ; thus inactive lanes are important and the immediate form cannot be used. define @orr_i32_ptrue_all_d( %a, %b) { ; CHECK-LABEL: orr_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: orr z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %pg.d = tail call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv2i1( %pg.d) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.orr.nxv4i32( %pg.s, %a, %b) ret %out } ; ; SQADD ; define @sqadd_i8( %a, %b) { ; CHECK-LABEL: sqadd_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.sqadd.nxv16i8( %pg, %a, %b) ret %out } define @sqadd_i16( %a, %b) { ; CHECK-LABEL: sqadd_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.sqadd.nxv8i16( %pg, %a, %b) ret %out } define @sqadd_i32( %a, %b) { ; CHECK-LABEL: sqadd_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.sqadd.nxv4i32( %pg, %a, %b) ret %out } define @sqadd_i64( %a, %b) { ; CHECK-LABEL: sqadd_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.sqadd.nxv2i64( %pg, %a, %b) ret %out } ; ; SQSUB ; define @sqsub_i8( %a, %b) { ; CHECK-LABEL: sqsub_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.sqsub.u.nxv16i8( %pg, %a, %b) ret %out } define @sqsub_i16( %a, %b) { ; CHECK-LABEL: sqsub_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.sqsub.u.nxv8i16( %pg, %a, %b) ret %out } define @sqsub_i32( %a, %b) { ; CHECK-LABEL: sqsub_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.sqsub.u.nxv4i32( %pg, %a, %b) ret %out } define @sqsub_i64( %a, %b) { ; CHECK-LABEL: sqsub_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.sqsub.u.nxv2i64( %pg, %a, %b) ret %out } ; ; UQADD ; define @uqadd_i8( %a, %b) { ; CHECK-LABEL: uqadd_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.uqadd.nxv16i8( %pg, %a, %b) ret %out } define @uqadd_i16( %a, %b) { ; CHECK-LABEL: uqadd_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.uqadd.nxv8i16( %pg, %a, %b) ret %out } define @uqadd_i32( %a, %b) { ; CHECK-LABEL: uqadd_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.uqadd.nxv4i32( %pg, %a, %b) ret %out } define @uqadd_i64( %a, %b) { ; CHECK-LABEL: uqadd_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.uqadd.nxv2i64( %pg, %a, %b) ret %out } ; ; UQSUB ; define @uqsub_i8( %a, %b) { ; CHECK-LABEL: uqsub_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.b, z0.b, z1.b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.uqsub.u.nxv16i8( %pg, %a, %b) ret %out } define @uqsub_i16( %a, %b) { ; CHECK-LABEL: uqsub_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.uqsub.u.nxv8i16( %pg, %a, %b) ret %out } define @uqsub_i32( %a, %b) { ; CHECK-LABEL: uqsub_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.uqsub.u.nxv4i32( %pg, %a, %b) ret %out } define @uqsub_i64( %a, %b) { ; CHECK-LABEL: uqsub_i64: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.uqsub.u.nxv2i64( %pg, %a, %b) ret %out } ; As uqsub_i32 but where pg is i8 based and thus compatible for i32. define @uqsub_i32_ptrue_all_b( %a, %b) { ; CHECK-LABEL: uqsub_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.b = tail call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.uqsub.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As uqsub_i32 but where pg is i16 based and thus compatible for i32. define @uqsub_i32_ptrue_all_h( %a, %b) { ; CHECK-LABEL: uqsub_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg.h = tail call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv8i1( %pg.h) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.uqsub.u.nxv4i32( %pg.s, %a, %b) ret %out } ; As uqsub_i32 but where pg is i64 based, which is not compatibile for i32 and ; thus inactive lanes are important and the immediate form cannot be used. define @uqsub_i32_ptrue_all_d( %a, %b) { ; CHECK-LABEL: uqsub_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: uqsub z0.s, p0/m, z0.s, z1.s ; CHECK-NEXT: ret %pg.d = tail call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv2i1( %pg.d) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.uqsub.nxv4i32( %pg.s, %a, %b) ret %out } ; ; ASR (wide) ; define @asr_i8( %a, %b) { ; CHECK-LABEL: asr_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: asr z0.b, z0.b, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.asr.wide.nxv16i8( %pg, %a, %b) ret %out } define @asr_i16( %a, %b) { ; CHECK-LABEL: asr_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: asr z0.h, z0.h, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.asr.wide.nxv8i16( %pg, %a, %b) ret %out } define @asr_i32( %a, %b) { ; CHECK-LABEL: asr_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: asr z0.s, z0.s, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.asr.wide.nxv4i32( %pg, %a, %b) ret %out } ; ; LSL (wide) ; define @lsl_i8( %a, %b) { ; CHECK-LABEL: lsl_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: lsl z0.b, z0.b, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.lsl.wide.nxv16i8( %pg, %a, %b) ret %out } define @lsl_i16( %a, %b) { ; CHECK-LABEL: lsl_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: lsl z0.h, z0.h, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.lsl.wide.nxv8i16( %pg, %a, %b) ret %out } define @lsl_i32( %a, %b) { ; CHECK-LABEL: lsl_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: lsl z0.s, z0.s, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.lsl.wide.nxv4i32( %pg, %a, %b) ret %out } ; ; LSR (wide) ; define @lsr_i8( %a, %b) { ; CHECK-LABEL: lsr_i8: ; CHECK: // %bb.0: ; CHECK-NEXT: lsr z0.b, z0.b, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %out = call @llvm.aarch64.sve.lsr.wide.nxv16i8( %pg, %a, %b) ret %out } define @lsr_i16( %a, %b) { ; CHECK-LABEL: lsr_i16: ; CHECK: // %bb.0: ; CHECK-NEXT: lsr z0.h, z0.h, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.lsr.wide.nxv8i16( %pg, %a, %b) ret %out } define @lsr_i32( %a, %b) { ; CHECK-LABEL: lsr_i32: ; CHECK: // %bb.0: ; CHECK-NEXT: lsr z0.s, z0.s, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.lsr.wide.nxv4i32( %pg, %a, %b) ret %out } ; As lsr_i32 but where pg is i8 based and thus compatible for i32. define @lsr_i32_ptrue_all_b( %a, %b) { ; CHECK-LABEL: lsr_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: lsr z0.s, z0.s, z1.d ; CHECK-NEXT: ret %pg.b = tail call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.lsr.wide.nxv4i32( %pg.s, %a, %b) ret %out } ; As lsr_i32 but where pg is i16 based and thus compatible for i32. define @lsr_i32_ptrue_all_h( %a, %b) { ; CHECK-LABEL: lsr_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: lsr z0.s, z0.s, z1.d ; CHECK-NEXT: ret %pg.h = tail call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv8i1( %pg.h) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.lsr.wide.nxv4i32( %pg.s, %a, %b) ret %out } ; As lsr_i32 but where pg is i64 based, which is not compatibile for i32 and ; thus inactive lanes are important and the immediate form cannot be used. define @lsr_i32_ptrue_all_d( %a, %b) { ; CHECK-LABEL: lsr_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: lsr z0.s, p0/m, z0.s, z1.d ; CHECK-NEXT: ret %pg.d = tail call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %pg.b = tail call @llvm.aarch64.sve.convert.to.svbool.nxv2i1( %pg.d) %pg.s = tail call @llvm.aarch64.sve.convert.from.svbool.nxv4i1( %pg.b) %out = tail call @llvm.aarch64.sve.lsr.wide.nxv4i32( %pg.s, %a, %b) ret %out } ; ; FADD ; define @fadd_half( %a, %b) { ; CHECK-LABEL: fadd_half: ; CHECK: // %bb.0: ; CHECK-NEXT: fadd z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.fadd.u.nxv8f16( %pg, %a, %b) ret %out } define @fadd_float( %a, %b) { ; CHECK-LABEL: fadd_float: ; CHECK: // %bb.0: ; CHECK-NEXT: fadd z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.fadd.u.nxv4f32( %pg, %a, %b) ret %out } define @fadd_double( %a, %b) { ; CHECK-LABEL: fadd_double: ; CHECK: // %bb.0: ; CHECK-NEXT: fadd z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.fadd.u.nxv2f64( %pg, %a, %b) ret %out } ; ; FSUB ; define @fsub_half( %a, %b) { ; CHECK-LABEL: fsub_half: ; CHECK: // %bb.0: ; CHECK-NEXT: fsub z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.fsub.u.nxv8f16( %pg, %a, %b) ret %out } define @fsub_float( %a, %b) { ; CHECK-LABEL: fsub_float: ; CHECK: // %bb.0: ; CHECK-NEXT: fsub z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.fsub.u.nxv4f32( %pg, %a, %b) ret %out } define @fsub_double( %a, %b) { ; CHECK-LABEL: fsub_double: ; CHECK: // %bb.0: ; CHECK-NEXT: fsub z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.fsub.u.nxv2f64( %pg, %a, %b) ret %out } ; ; FMUL ; define @fmul_half( %a, %b) { ; CHECK-LABEL: fmul_half: ; CHECK: // %bb.0: ; CHECK-NEXT: fmul z0.h, z0.h, z1.h ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %out = call @llvm.aarch64.sve.fmul.u.nxv8f16( %pg, %a, %b) ret %out } define @fmul_float( %a, %b) { ; CHECK-LABEL: fmul_float: ; CHECK: // %bb.0: ; CHECK-NEXT: fmul z0.s, z0.s, z1.s ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %out = call @llvm.aarch64.sve.fmul.u.nxv4f32( %pg, %a, %b) ret %out } define @fmul_double( %a, %b) { ; CHECK-LABEL: fmul_double: ; CHECK: // %bb.0: ; CHECK-NEXT: fmul z0.d, z0.d, z1.d ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %out = call @llvm.aarch64.sve.fmul.u.nxv2f64( %pg, %a, %b) ret %out } declare @llvm.aarch64.sve.add.u.nxv16i8(, , ) declare @llvm.aarch64.sve.add.u.nxv8i16(, , ) declare @llvm.aarch64.sve.add.u.nxv4i32(, , ) declare @llvm.aarch64.sve.add.u.nxv2i64(, , ) declare @llvm.aarch64.sve.sub.nxv4i32(, , ) declare @llvm.aarch64.sve.sub.u.nxv16i8(, , ) declare @llvm.aarch64.sve.sub.u.nxv8i16(, , ) declare @llvm.aarch64.sve.sub.u.nxv4i32(, , ) declare @llvm.aarch64.sve.sub.u.nxv2i64(, , ) declare @llvm.aarch64.sve.mul.u.nxv16i8(, , ) declare @llvm.aarch64.sve.mul.u.nxv8i16(, , ) declare @llvm.aarch64.sve.mul.u.nxv4i32(, , ) declare @llvm.aarch64.sve.mul.u.nxv2i64(, , ) declare @llvm.aarch64.sve.smulh.u.nxv16i8(, , ) declare @llvm.aarch64.sve.smulh.u.nxv8i16(, , ) declare @llvm.aarch64.sve.smulh.u.nxv4i32(, , ) declare @llvm.aarch64.sve.smulh.u.nxv2i64(, , ) declare @llvm.aarch64.sve.umulh.nxv4i32(, , ) declare @llvm.aarch64.sve.umulh.u.nxv16i8(, , ) declare @llvm.aarch64.sve.umulh.u.nxv8i16(, , ) declare @llvm.aarch64.sve.umulh.u.nxv4i32(, , ) declare @llvm.aarch64.sve.umulh.u.nxv2i64(, , ) declare @llvm.aarch64.sve.and.u.nxv16i8(, , ) declare @llvm.aarch64.sve.and.u.nxv8i16(, , ) declare @llvm.aarch64.sve.and.u.nxv4i32(, , ) declare @llvm.aarch64.sve.and.u.nxv2i64(, , ) declare @llvm.aarch64.sve.bic.u.nxv16i8(, , ) declare @llvm.aarch64.sve.bic.u.nxv8i16(, , ) declare @llvm.aarch64.sve.bic.u.nxv4i32(, , ) declare @llvm.aarch64.sve.bic.u.nxv2i64(, , ) declare @llvm.aarch64.sve.eor.u.nxv16i8(, , ) declare @llvm.aarch64.sve.eor.u.nxv8i16(, , ) declare @llvm.aarch64.sve.eor.u.nxv4i32(, , ) declare @llvm.aarch64.sve.eor.u.nxv2i64(, , ) declare @llvm.aarch64.sve.orr.nxv4i32(, , ) declare @llvm.aarch64.sve.orr.u.nxv16i8(, , ) declare @llvm.aarch64.sve.orr.u.nxv8i16(, , ) declare @llvm.aarch64.sve.orr.u.nxv4i32(, , ) declare @llvm.aarch64.sve.orr.u.nxv2i64(, , ) declare @llvm.aarch64.sve.sqadd.nxv16i8(, , ) declare @llvm.aarch64.sve.sqadd.nxv8i16(, , ) declare @llvm.aarch64.sve.sqadd.nxv4i32(, , ) declare @llvm.aarch64.sve.sqadd.nxv2i64(, , ) declare @llvm.aarch64.sve.sqsub.u.nxv16i8(, , ) declare @llvm.aarch64.sve.sqsub.u.nxv8i16(, , ) declare @llvm.aarch64.sve.sqsub.u.nxv4i32(, , ) declare @llvm.aarch64.sve.sqsub.u.nxv2i64(, , ) declare @llvm.aarch64.sve.uqadd.nxv16i8(, , ) declare @llvm.aarch64.sve.uqadd.nxv8i16(, , ) declare @llvm.aarch64.sve.uqadd.nxv4i32(, , ) declare @llvm.aarch64.sve.uqadd.nxv2i64(, , ) declare @llvm.aarch64.sve.uqsub.nxv4i32(, , ) declare @llvm.aarch64.sve.uqsub.u.nxv16i8(, , ) declare @llvm.aarch64.sve.uqsub.u.nxv8i16(, , ) declare @llvm.aarch64.sve.uqsub.u.nxv4i32(, , ) declare @llvm.aarch64.sve.uqsub.u.nxv2i64(, , ) declare @llvm.aarch64.sve.asr.wide.nxv16i8(, , ) declare @llvm.aarch64.sve.asr.wide.nxv8i16(, , ) declare @llvm.aarch64.sve.asr.wide.nxv4i32(, , ) declare @llvm.aarch64.sve.lsl.wide.nxv16i8(, , ) declare @llvm.aarch64.sve.lsl.wide.nxv8i16(, , ) declare @llvm.aarch64.sve.lsl.wide.nxv4i32(, , ) declare @llvm.aarch64.sve.lsr.wide.nxv16i8(, , ) declare @llvm.aarch64.sve.lsr.wide.nxv8i16(, , ) declare @llvm.aarch64.sve.lsr.wide.nxv4i32(, , ) declare @llvm.aarch64.sve.fadd.u.nxv8f16(, , ) declare @llvm.aarch64.sve.fadd.u.nxv4f32(, , ) declare @llvm.aarch64.sve.fadd.u.nxv2f64(, , ) declare @llvm.aarch64.sve.fsub.u.nxv8f16(, , ) declare @llvm.aarch64.sve.fsub.u.nxv4f32(, , ) declare @llvm.aarch64.sve.fsub.u.nxv2f64(, , ) declare @llvm.aarch64.sve.fmul.u.nxv8f16(, , ) declare @llvm.aarch64.sve.fmul.u.nxv4f32(, , ) declare @llvm.aarch64.sve.fmul.u.nxv2f64(, , ) declare @llvm.aarch64.sve.convert.from.svbool.nxv4i1() declare @llvm.aarch64.sve.convert.from.svbool.nxv8i1() declare @llvm.aarch64.sve.convert.from.svbool.nxv2i1() declare @llvm.aarch64.sve.convert.to.svbool.nxv4i1() declare @llvm.aarch64.sve.convert.to.svbool.nxv8i1() declare @llvm.aarch64.sve.convert.to.svbool.nxv2i1() declare @llvm.aarch64.sve.ptrue.nxv16i1(i32) declare @llvm.aarch64.sve.ptrue.nxv8i1(i32) declare @llvm.aarch64.sve.ptrue.nxv4i1(i32) declare @llvm.aarch64.sve.ptrue.nxv2i1(i32)