; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=riscv32 -mattr=+v -verify-machineinstrs < %s | FileCheck %s ; RUN: llc -mtriple=riscv64 -mattr=+v -verify-machineinstrs < %s | FileCheck %s ; Test that the prepareSREMEqFold optimization doesn't crash on scalable ; vector types. define @srem_eq_fold_nxv4i8( %va) { ; CHECK-LABEL: srem_eq_fold_nxv4i8: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, 42 ; CHECK-NEXT: vsetvli a1, zero, e8, mf2, ta, ma ; CHECK-NEXT: vmv.v.x v9, a0 ; CHECK-NEXT: li a1, -85 ; CHECK-NEXT: vmacc.vx v9, a1, v8 ; CHECK-NEXT: vsll.vi v8, v9, 7 ; CHECK-NEXT: vsrl.vi v9, v9, 1 ; CHECK-NEXT: vor.vv v8, v9, v8 ; CHECK-NEXT: vmsleu.vx v0, v8, a0 ; CHECK-NEXT: ret %head_six = insertelement poison, i8 6, i32 0 %splat_six = shufflevector %head_six, poison, zeroinitializer %rem = srem %va, %splat_six %cc = icmp eq %rem, zeroinitializer ret %cc } define @vmulh_vv_nxv1i32( %va, %vb) { ; CHECK-LABEL: vmulh_vv_nxv1i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32, mf2, ta, ma ; CHECK-NEXT: vmulh.vv v8, v9, v8 ; CHECK-NEXT: ret %vc = sext %vb to %vd = sext %va to %ve = mul %vc, %vd %head = insertelement poison, i64 32, i32 0 %splat = shufflevector %head, poison, zeroinitializer %vf = lshr %ve, %splat %vg = trunc %vf to ret %vg } define @vmulh_vx_nxv1i32( %va, i32 %x) { ; CHECK-LABEL: vmulh_vx_nxv1i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 %x, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv1i32_0( %va) { ; CHECK-LABEL: vmulh_vi_nxv1i32_0: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, -7 ; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 -7, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv1i32_1( %va) { ; CHECK-LABEL: vmulh_vi_nxv1i32_1: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, 16 ; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 16, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vv_nxv2i32( %va, %vb) { ; CHECK-LABEL: vmulh_vv_nxv2i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, ma ; CHECK-NEXT: vmulh.vv v8, v9, v8 ; CHECK-NEXT: ret %vc = sext %vb to %vd = sext %va to %ve = mul %vc, %vd %head = insertelement poison, i64 32, i32 0 %splat = shufflevector %head, poison, zeroinitializer %vf = lshr %ve, %splat %vg = trunc %vf to ret %vg } define @vmulh_vx_nxv2i32( %va, i32 %x) { ; CHECK-LABEL: vmulh_vx_nxv2i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 %x, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv2i32_0( %va) { ; CHECK-LABEL: vmulh_vi_nxv2i32_0: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, -7 ; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 -7, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv2i32_1( %va) { ; CHECK-LABEL: vmulh_vi_nxv2i32_1: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, 16 ; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 16, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vv_nxv4i32( %va, %vb) { ; CHECK-LABEL: vmulh_vv_nxv4i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32, m2, ta, ma ; CHECK-NEXT: vmulh.vv v8, v10, v8 ; CHECK-NEXT: ret %vc = sext %vb to %vd = sext %va to %ve = mul %vc, %vd %head = insertelement poison, i64 32, i32 0 %splat = shufflevector %head, poison, zeroinitializer %vf = lshr %ve, %splat %vg = trunc %vf to ret %vg } define @vmulh_vx_nxv4i32( %va, i32 %x) { ; CHECK-LABEL: vmulh_vx_nxv4i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 %x, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv4i32_0( %va) { ; CHECK-LABEL: vmulh_vi_nxv4i32_0: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, -7 ; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 -7, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv4i32_1( %va) { ; CHECK-LABEL: vmulh_vi_nxv4i32_1: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, 16 ; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 16, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vv_nxv8i32( %va, %vb) { ; CHECK-LABEL: vmulh_vv_nxv8i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32, m4, ta, ma ; CHECK-NEXT: vmulh.vv v8, v12, v8 ; CHECK-NEXT: ret %vc = sext %vb to %vd = sext %va to %ve = mul %vc, %vd %head = insertelement poison, i64 32, i32 0 %splat = shufflevector %head, poison, zeroinitializer %vf = lshr %ve, %splat %vg = trunc %vf to ret %vg } define @vmulh_vx_nxv8i32( %va, i32 %x) { ; CHECK-LABEL: vmulh_vx_nxv8i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 %x, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv8i32_0( %va) { ; CHECK-LABEL: vmulh_vi_nxv8i32_0: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, -7 ; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 -7, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf } define @vmulh_vi_nxv8i32_1( %va) { ; CHECK-LABEL: vmulh_vi_nxv8i32_1: ; CHECK: # %bb.0: ; CHECK-NEXT: li a0, 16 ; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, ma ; CHECK-NEXT: vmulh.vx v8, v8, a0 ; CHECK-NEXT: ret %head1 = insertelement poison, i32 16, i32 0 %splat1 = shufflevector %head1, poison, zeroinitializer %vb = sext %splat1 to %vc = sext %va to %vd = mul %vb, %vc %head2 = insertelement poison, i64 32, i32 0 %splat2 = shufflevector %head2, poison, zeroinitializer %ve = lshr %vd, %splat2 %vf = trunc %ve to ret %vf }