; RUN: llc -mtriple=aarch64-linux-gnu -mattr=+sme2 -verify-machineinstrs < %s | FileCheck %s ; SQDMULH (Single, x2) define { , } @multi_vec_sat_double_mulh_single_x2_s8( %unused, %zdn1, %zdn2, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x2_s8: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.b, z5.b }, { z4.b, z5.b }, z3.b ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv16i8( %zdn1, %zdn2, %zm) ret { , } %res } define { , } @multi_vec_sat_double_mulh_single_x2_s16( %unused, %zdn1, %zdn2, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x2_s16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.h, z5.h }, { z4.h, z5.h }, z3.h ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv8i16( %zdn1, %zdn2, %zm) ret { , } %res } define { , } @multi_vec_sat_double_mulh_single_x2_s32( %unused, %zdn1, %zdn2, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x2_s32: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.s, z5.s }, { z4.s, z5.s }, z3.s ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv4i32( %zdn1, %zdn2, %zm) ret { , } %res } define { , } @multi_vec_sat_double_mulh_single_x2_s64( %unused, %zdn1, %zdn2, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x2_s64: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.d, z5.d }, { z4.d, z5.d }, z3.d ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv2i64( %zdn1, %zdn2, %zm) ret { , } %res } ; SQDMULH (Single, x4) define { , , , } @multi_vec_sat_double_mulh_single_x4_s8( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x4_s8: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: sqdmulh { z24.b - z27.b }, { z24.b - z27.b }, z5.b ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv16i8( %zdn1, %zdn2, %zdn3, %zdn4, %zm) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_single_x4_s16( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x4_s16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: sqdmulh { z24.h - z27.h }, { z24.h - z27.h }, z5.h ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv8i16( %zdn1, %zdn2, %zdn3, %zdn4, %zm) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_single_x4_s32( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x4_s32: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: sqdmulh { z24.s - z27.s }, { z24.s - z27.s }, z5.s ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv4i32( %zdn1, %zdn2, %zdn3, %zdn4, %zm) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_single_x4_s64( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm) { ; CHECK-LABEL: multi_vec_sat_double_mulh_single_x4_s64: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: sqdmulh { z24.d - z27.d }, { z24.d - z27.d }, z5.d ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv2i64( %zdn1, %zdn2, %zdn3, %zdn4, %zm) ret { , , , } %res } ; SQDMULH (x2, Multi) define { , } @multi_vec_sat_double_mulh_multi_x2_s8( %unused, %zdn1, %zdn2, %zm1, %zm2) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x2_s8: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z7.d, z4.d ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z6.d, z3.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.b, z5.b }, { z4.b, z5.b }, { z6.b, z7.b } ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv16i8( %zdn1, %zdn2, %zm1, %zm2) ret { , } %res } define { , } @multi_vec_sat_double_mulh_multi_x2_s16( %unused, %zdn1, %zdn2, %zm1, %zm2) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x2_s16: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z7.d, z4.d ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z6.d, z3.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.h, z5.h }, { z4.h, z5.h }, { z6.h, z7.h } ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv8i16( %zdn1, %zdn2, %zm1, %zm2) ret { , } %res } define { , } @multi_vec_sat_double_mulh_multi_x2_s32( %unused, %zdn1, %zdn2, %zm1, %zm2) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x2_s32: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z7.d, z4.d ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z6.d, z3.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.s, z5.s }, { z4.s, z5.s }, { z6.s, z7.s } ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv4i32( %zdn1, %zdn2, %zm1, %zm2) ret { , } %res } define { , } @multi_vec_sat_double_mulh_multi_x2_s64( %unused, %zdn1, %zdn2, %zm1, %zm2) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x2_s64: ; CHECK: // %bb.0: ; CHECK-NEXT: mov z7.d, z4.d ; CHECK-NEXT: mov z5.d, z2.d ; CHECK-NEXT: mov z6.d, z3.d ; CHECK-NEXT: mov z4.d, z1.d ; CHECK-NEXT: sqdmulh { z4.d, z5.d }, { z4.d, z5.d }, { z6.d, z7.d } ; CHECK-NEXT: mov z0.d, z4.d ; CHECK-NEXT: mov z1.d, z5.d ; CHECK-NEXT: ret %res = call { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv2i64( %zdn1, %zdn2, %zm1, %zm2) ret { , } %res } ; SQDMULH (x4, Multi) define { , , , } @multi_vec_sat_double_mulh_multi_x4_s8( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x4_s8: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.b ; CHECK-NEXT: mov z30.d, z7.d ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z29.d, z6.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z28.d, z5.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: ld1b { z31.b }, p0/z, [x0] ; CHECK-NEXT: sqdmulh { z24.b - z27.b }, { z24.b - z27.b }, { z28.b - z31.b } ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv16i8( %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_multi_x4_s16( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x4_s16: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.h ; CHECK-NEXT: mov z30.d, z7.d ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z29.d, z6.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z28.d, z5.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: ld1h { z31.h }, p0/z, [x0] ; CHECK-NEXT: sqdmulh { z24.h - z27.h }, { z24.h - z27.h }, { z28.h - z31.h } ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv8i16( %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_multi_x4_s32( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x4_s32: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.s ; CHECK-NEXT: mov z30.d, z7.d ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z29.d, z6.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z28.d, z5.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: ld1w { z31.s }, p0/z, [x0] ; CHECK-NEXT: sqdmulh { z24.s - z27.s }, { z24.s - z27.s }, { z28.s - z31.s } ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv4i32( %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) ret { , , , } %res } define { , , , } @multi_vec_sat_double_mulh_multi_x4_s64( %unused, %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) { ; CHECK-LABEL: multi_vec_sat_double_mulh_multi_x4_s64: ; CHECK: // %bb.0: ; CHECK-NEXT: ptrue p0.d ; CHECK-NEXT: mov z30.d, z7.d ; CHECK-NEXT: mov z27.d, z4.d ; CHECK-NEXT: mov z29.d, z6.d ; CHECK-NEXT: mov z26.d, z3.d ; CHECK-NEXT: mov z28.d, z5.d ; CHECK-NEXT: mov z25.d, z2.d ; CHECK-NEXT: mov z24.d, z1.d ; CHECK-NEXT: ld1d { z31.d }, p0/z, [x0] ; CHECK-NEXT: sqdmulh { z24.d - z27.d }, { z24.d - z27.d }, { z28.d - z31.d } ; CHECK-NEXT: mov z0.d, z24.d ; CHECK-NEXT: mov z1.d, z25.d ; CHECK-NEXT: mov z2.d, z26.d ; CHECK-NEXT: mov z3.d, z27.d ; CHECK-NEXT: ret %res = call { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv2i64( %zdn1, %zdn2, %zdn3, %zdn4, %zm1, %zm2, %zm3, %zm4) ret { , , , } %res } declare { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv16i8(, , ) declare { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv8i16(, , ) declare { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv4i32(, , ) declare { , } @llvm.aarch64.sve.sqdmulh.single.vgx2.nxv2i64(, , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv16i8(, , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv8i16(, , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv4i32(, , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.single.vgx4.nxv2i64(, , , , ) declare { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv16i8(, , , ) declare { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv8i16(, , , ) declare { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv4i32(, , , ) declare { , } @llvm.aarch64.sve.sqdmulh.vgx2.nxv2i64(, , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv16i8(, , , , , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv8i16(, , , , , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv4i32(, , , , , , , ) declare { , , , } @llvm.aarch64.sve.sqdmulh.vgx4.nxv2i64(, , , , , , , )