; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sve2 < %s | FileCheck %s ; SQADD define @sqadd_b_lowimm( %a) { ; CHECK-LABEL: sqadd_b_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.b, z0.b, #27 // =0x1b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %elt = insertelement undef, i8 27, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv16i8( %pg, %a, %splat) ret %out } define @sqadd_h_lowimm( %a) { ; CHECK-LABEL: sqadd_h_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.h, z0.h, #43 // =0x2b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 43, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv8i16( %pg, %a, %splat) ret %out } define @sqadd_h_highimm( %a) { ; CHECK-LABEL: sqadd_h_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.h, z0.h, #2048 // =0x800 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 2048, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv8i16( %pg, %a, %splat) ret %out } define @sqadd_s_lowimm( %a) { ; CHECK-LABEL: sqadd_s_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.s, z0.s, #1 // =0x1 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 1, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv4i32( %pg, %a, %splat) ret %out } define @sqadd_s_highimm( %a) { ; CHECK-LABEL: sqadd_s_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.s, z0.s, #8192 // =0x2000 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 8192, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv4i32( %pg, %a, %splat) ret %out } define @sqadd_d_lowimm( %a) { ; CHECK-LABEL: sqadd_d_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.d, z0.d, #255 // =0xff ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 255, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv2i64( %pg, %a, %splat) ret %out } define @sqadd_d_highimm( %a) { ; CHECK-LABEL: sqadd_d_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqadd z0.d, z0.d, #65280 // =0xff00 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 65280, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqadd.nxv2i64( %pg, %a, %splat) ret %out } ; SQSUB define @sqsub_b_lowimm( %a) { ; CHECK-LABEL: sqsub_b_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.b, z0.b, #27 // =0x1b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %elt = insertelement undef, i8 27, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv16i8( %pg, %a, %splat) ret %out } define @sqsub_h_lowimm( %a) { ; CHECK-LABEL: sqsub_h_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.h, z0.h, #43 // =0x2b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 43, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv8i16( %pg, %a, %splat) ret %out } define @sqsub_h_highimm( %a) { ; CHECK-LABEL: sqsub_h_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.h, z0.h, #2048 // =0x800 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 2048, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv8i16( %pg, %a, %splat) ret %out } define @sqsub_s_lowimm( %a) { ; CHECK-LABEL: sqsub_s_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.s, z0.s, #1 // =0x1 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 1, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv4i32( %pg, %a, %splat) ret %out } define @sqsub_s_highimm( %a) { ; CHECK-LABEL: sqsub_s_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.s, z0.s, #8192 // =0x2000 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 8192, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv4i32( %pg, %a, %splat) ret %out } define @sqsub_d_lowimm( %a) { ; CHECK-LABEL: sqsub_d_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.d, z0.d, #255 // =0xff ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 255, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv2i64( %pg, %a, %splat) ret %out } define @sqsub_d_highimm( %a) { ; CHECK-LABEL: sqsub_d_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: sqsub z0.d, z0.d, #65280 // =0xff00 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 65280, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.sqsub.u.nxv2i64( %pg, %a, %splat) ret %out } ; UQADD define @uqadd_b_lowimm( %a) { ; CHECK-LABEL: uqadd_b_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.b, z0.b, #27 // =0x1b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %elt = insertelement undef, i8 27, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv16i8( %pg, %a, %splat) ret %out } define @uqadd_h_lowimm( %a) { ; CHECK-LABEL: uqadd_h_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.h, z0.h, #43 // =0x2b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 43, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv8i16( %pg, %a, %splat) ret %out } define @uqadd_h_highimm( %a) { ; CHECK-LABEL: uqadd_h_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.h, z0.h, #2048 // =0x800 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 2048, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv8i16( %pg, %a, %splat) ret %out } define @uqadd_s_lowimm( %a) { ; CHECK-LABEL: uqadd_s_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.s, z0.s, #1 // =0x1 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 1, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv4i32( %pg, %a, %splat) ret %out } define @uqadd_s_highimm( %a) { ; CHECK-LABEL: uqadd_s_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.s, z0.s, #8192 // =0x2000 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 8192, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv4i32( %pg, %a, %splat) ret %out } define @uqadd_d_lowimm( %a) { ; CHECK-LABEL: uqadd_d_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.d, z0.d, #255 // =0xff ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 255, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv2i64( %pg, %a, %splat) ret %out } define @uqadd_d_highimm( %a) { ; CHECK-LABEL: uqadd_d_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqadd z0.d, z0.d, #65280 // =0xff00 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 65280, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqadd.nxv2i64( %pg, %a, %splat) ret %out } ; UQSUB define @uqsub_b_lowimm( %a) { ; CHECK-LABEL: uqsub_b_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.b, z0.b, #27 // =0x1b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv16i1(i32 31) %elt = insertelement undef, i8 27, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv16i8( %pg, %a, %splat) ret %out } define @uqsub_h_lowimm( %a) { ; CHECK-LABEL: uqsub_h_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.h, z0.h, #43 // =0x2b ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 43, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv8i16( %pg, %a, %splat) ret %out } define @uqsub_h_highimm( %a) { ; CHECK-LABEL: uqsub_h_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.h, z0.h, #2048 // =0x800 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv8i1(i32 31) %elt = insertelement undef, i16 2048, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv8i16( %pg, %a, %splat) ret %out } define @uqsub_s_lowimm( %a) { ; CHECK-LABEL: uqsub_s_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, #1 // =0x1 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 1, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv4i32( %pg, %a, %splat) ret %out } define @uqsub_s_highimm( %a) { ; CHECK-LABEL: uqsub_s_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, #8192 // =0x2000 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv4i1(i32 31) %elt = insertelement undef, i32 8192, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv4i32( %pg, %a, %splat) ret %out } define @uqsub_d_lowimm( %a) { ; CHECK-LABEL: uqsub_d_lowimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.d, z0.d, #255 // =0xff ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 255, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv2i64( %pg, %a, %splat) ret %out } define @uqsub_d_highimm( %a) { ; CHECK-LABEL: uqsub_d_highimm: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.d, z0.d, #65280 // =0xff00 ; CHECK-NEXT: ret %pg = call @llvm.aarch64.sve.ptrue.nxv2i1(i32 31) %elt = insertelement undef, i64 65280, i32 0 %splat = shufflevector %elt, undef, zeroinitializer %out = call @llvm.aarch64.sve.uqsub.u.nxv2i64( %pg, %a, %splat) ret %out } ; As uqsub_i32 but where pg is i8 based and thus compatible for i32. define @uqsub_i32_ptrue_all_b( %a) #0 { ; CHECK-LABEL: uqsub_i32_ptrue_all_b: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, #1 // =0x1 ; 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) %b = tail call @llvm.aarch64.sve.dup.x.nxv4i32(i32 1) %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) #0 { ; CHECK-LABEL: uqsub_i32_ptrue_all_h: ; CHECK: // %bb.0: ; CHECK-NEXT: uqsub z0.s, z0.s, #1 // =0x1 ; 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) %b = tail call @llvm.aarch64.sve.dup.x.nxv4i32(i32 1) %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) #0 { ; CHECK-LABEL: uqsub_i32_ptrue_all_d: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z1.s, #1 // =0x1 ; 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) %b = tail call @llvm.aarch64.sve.dup.x.nxv4i32(i32 1) %out = tail call @llvm.aarch64.sve.uqsub.nxv4i32( %pg.s, %a, %b) ret %out } 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.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.dup.x.nxv4i32(i32) declare @llvm.aarch64.sve.ptrue.nxv16i1(i32 %pattern) declare @llvm.aarch64.sve.ptrue.nxv8i1(i32 %pattern) declare @llvm.aarch64.sve.ptrue.nxv4i1(i32 %pattern) declare @llvm.aarch64.sve.ptrue.nxv2i1(i32 %pattern)