diff --git a/llvm/test/CodeGen/RISCV/rvv/vwsub-sdnode.ll b/llvm/test/CodeGen/RISCV/rvv/vwsub-sdnode.ll new file mode 100644 index 0000000000000..36f9305504aa8 --- /dev/null +++ b/llvm/test/CodeGen/RISCV/rvv/vwsub-sdnode.ll @@ -0,0 +1,427 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; RUN: llc -mtriple=riscv32 -mattr=+experimental-v -verify-machineinstrs < %s | FileCheck %s +; RUN: llc -mtriple=riscv64 -mattr=+experimental-v -verify-machineinstrs < %s | FileCheck %s + +define @vwsub_vv_nxv1i64( %va, %vb) { +; CHECK-LABEL: vwsub_vv_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsub.vv v10, v8, v9 +; CHECK-NEXT: vmv1r.v v8, v10 +; CHECK-NEXT: ret + %vc = sext %va to + %vd = sext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vv_nxv1i64( %va, %vb) { +; CHECK-LABEL: vwsubu_vv_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsubu.vv v10, v8, v9 +; CHECK-NEXT: vmv1r.v v8, v10 +; CHECK-NEXT: ret + %vc = zext %va to + %vd = zext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_vx_nxv1i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_vx_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsub.vx v9, v8, a0 +; CHECK-NEXT: vmv1r.v v8, v9 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = sext %va to + %vd = sext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vx_nxv1i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_vx_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsubu.vx v9, v8, a0 +; CHECK-NEXT: vmv1r.v v8, v9 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = zext %va to + %vd = zext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_wv_nxv1i64( %va, %vb) { +; CHECK-LABEL: vwsub_wv_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsub.wv v10, v8, v9 +; CHECK-NEXT: vmv1r.v v8, v10 +; CHECK-NEXT: ret + %vc = sext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsubu_wv_nxv1i64( %va, %vb) { +; CHECK-LABEL: vwsubu_wv_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsubu.wv v10, v8, v9 +; CHECK-NEXT: vmv1r.v v8, v10 +; CHECK-NEXT: ret + %vc = zext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsub_wx_nxv1i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_wx_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsub.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = sext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsubu_wx_nxv1i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_wx_nxv1i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, mf2, ta, mu +; CHECK-NEXT: vwsubu.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = zext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsub_vv_nxv2i64( %va, %vb) { +; CHECK-LABEL: vwsub_vv_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsub.vv v10, v8, v9 +; CHECK-NEXT: vmv2r.v v8, v10 +; CHECK-NEXT: ret + %vc = sext %va to + %vd = sext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vv_nxv2i64( %va, %vb) { +; CHECK-LABEL: vwsubu_vv_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsubu.vv v10, v8, v9 +; CHECK-NEXT: vmv2r.v v8, v10 +; CHECK-NEXT: ret + %vc = zext %va to + %vd = zext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_vx_nxv2i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_vx_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsub.vx v10, v8, a0 +; CHECK-NEXT: vmv2r.v v8, v10 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = sext %va to + %vd = sext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vx_nxv2i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_vx_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsubu.vx v10, v8, a0 +; CHECK-NEXT: vmv2r.v v8, v10 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = zext %va to + %vd = zext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_wv_nxv2i64( %va, %vb) { +; CHECK-LABEL: vwsub_wv_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsub.wv v12, v8, v10 +; CHECK-NEXT: vmv2r.v v8, v12 +; CHECK-NEXT: ret + %vc = sext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsubu_wv_nxv2i64( %va, %vb) { +; CHECK-LABEL: vwsubu_wv_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsubu.wv v12, v8, v10 +; CHECK-NEXT: vmv2r.v v8, v12 +; CHECK-NEXT: ret + %vc = zext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsub_wx_nxv2i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_wx_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsub.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = sext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsubu_wx_nxv2i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_wx_nxv2i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m1, ta, mu +; CHECK-NEXT: vwsubu.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = zext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsub_vv_nxv4i64( %va, %vb) { +; CHECK-LABEL: vwsub_vv_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsub.vv v12, v8, v10 +; CHECK-NEXT: vmv4r.v v8, v12 +; CHECK-NEXT: ret + %vc = sext %va to + %vd = sext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vv_nxv4i64( %va, %vb) { +; CHECK-LABEL: vwsubu_vv_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsubu.vv v12, v8, v10 +; CHECK-NEXT: vmv4r.v v8, v12 +; CHECK-NEXT: ret + %vc = zext %va to + %vd = zext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_vx_nxv4i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_vx_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsub.vx v12, v8, a0 +; CHECK-NEXT: vmv4r.v v8, v12 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = sext %va to + %vd = sext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vx_nxv4i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_vx_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsubu.vx v12, v8, a0 +; CHECK-NEXT: vmv4r.v v8, v12 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = zext %va to + %vd = zext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_wv_nxv4i64( %va, %vb) { +; CHECK-LABEL: vwsub_wv_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsub.wv v16, v8, v12 +; CHECK-NEXT: vmv4r.v v8, v16 +; CHECK-NEXT: ret + %vc = sext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsubu_wv_nxv4i64( %va, %vb) { +; CHECK-LABEL: vwsubu_wv_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsubu.wv v16, v8, v12 +; CHECK-NEXT: vmv4r.v v8, v16 +; CHECK-NEXT: ret + %vc = zext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsub_wx_nxv4i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_wx_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsub.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = sext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsubu_wx_nxv4i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_wx_nxv4i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m2, ta, mu +; CHECK-NEXT: vwsubu.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = zext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsub_vv_nxv8i64( %va, %vb) { +; CHECK-LABEL: vwsub_vv_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsub.vv v16, v8, v12 +; CHECK-NEXT: vmv8r.v v8, v16 +; CHECK-NEXT: ret + %vc = sext %va to + %vd = sext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vv_nxv8i64( %va, %vb) { +; CHECK-LABEL: vwsubu_vv_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsubu.vv v16, v8, v12 +; CHECK-NEXT: vmv8r.v v8, v16 +; CHECK-NEXT: ret + %vc = zext %va to + %vd = zext %vb to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_vx_nxv8i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_vx_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsub.vx v16, v8, a0 +; CHECK-NEXT: vmv8r.v v8, v16 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = sext %va to + %vd = sext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsubu_vx_nxv8i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_vx_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsubu.vx v16, v8, a0 +; CHECK-NEXT: vmv8r.v v8, v16 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vc = zext %va to + %vd = zext %splat to + %ve = sub %vc, %vd + ret %ve +} + +define @vwsub_wv_nxv8i64( %va, %vb) { +; CHECK-LABEL: vwsub_wv_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsub.wv v24, v8, v16 +; CHECK-NEXT: vmv8r.v v8, v24 +; CHECK-NEXT: ret + %vc = sext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsubu_wv_nxv8i64( %va, %vb) { +; CHECK-LABEL: vwsubu_wv_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a0, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsubu.wv v24, v8, v16 +; CHECK-NEXT: vmv8r.v v8, v24 +; CHECK-NEXT: ret + %vc = zext %vb to + %vd = sub %va, %vc + ret %vd +} + +define @vwsub_wx_nxv8i64( %va, i32 %b) { +; CHECK-LABEL: vwsub_wx_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsub.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = sext %splat to + %vc = sub %va, %vb + ret %vc +} + +define @vwsubu_wx_nxv8i64( %va, i32 %b) { +; CHECK-LABEL: vwsubu_wx_nxv8i64: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetvli a1, zero, e32, m4, ta, mu +; CHECK-NEXT: vwsubu.wx v8, v8, a0 +; CHECK-NEXT: ret + %head = insertelement undef, i32 %b, i32 0 + %splat = shufflevector %head, undef, zeroinitializer + %vb = zext %splat to + %vc = sub %va, %vb + ret %vc +}