From c5c6ea8e12b623880597963ce7274223b47ac5d2 Mon Sep 17 00:00:00 2001
From: Philip Reames <preames@rivosinc.com>
Date: Tue, 16 May 2023 12:39:48 -0700
Subject: [PATCH] [RISCV] Precommit coverage for an upcoming dag combine change

---
 .../CodeGen/RISCV/rvv/fixed-vectors-int-splat.ll   | 113 +++++++++++++++++++++
 1 file changed, 113 insertions(+)

diff --git a/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-int-splat.ll b/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-int-splat.ll
index cf43b92..5b5c94f 100644
--- a/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-int-splat.ll
+++ b/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-int-splat.ll
@@ -447,6 +447,119 @@ define void @splat_zero_v4i64(ptr %x) {
   ret void
 }
 
+define void @splat_zero_v2i16(ptr %p) {
+; CHECK-LABEL: splat_zero_v2i16:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetivli zero, 2, e16, mf4, ta, ma
+; CHECK-NEXT:    vmv.v.i v8, 0
+; CHECK-NEXT:    vse16.v v8, (a0)
+; CHECK-NEXT:    ret
+  store <2 x i16> zeroinitializer, ptr %p
+  ret void
+}
+
+define void @splat_zero_v2i16_unaligned(ptr %p) {
+; CHECK-LABEL: splat_zero_v2i16_unaligned:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetivli zero, 2, e16, mf4, ta, ma
+; CHECK-NEXT:    vmv.v.i v8, 0
+; CHECK-NEXT:    vsetivli zero, 4, e8, mf4, ta, ma
+; CHECK-NEXT:    vse8.v v8, (a0)
+; CHECK-NEXT:    ret
+  store <2 x i16> zeroinitializer, ptr %p, align 1
+  ret void
+}
+
+define void @splat_zero_v4i16(ptr %p) {
+; CHECK-LABEL: splat_zero_v4i16:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetivli zero, 4, e16, mf2, ta, ma
+; CHECK-NEXT:    vmv.v.i v8, 0
+; CHECK-NEXT:    vse16.v v8, (a0)
+; CHECK-NEXT:    ret
+  store <4 x i16> zeroinitializer, ptr %p
+  ret void
+}
+
+define void @splat_zero_v2i32(ptr %p) {
+; CHECK-LABEL: splat_zero_v2i32:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetivli zero, 2, e32, mf2, ta, ma
+; CHECK-NEXT:    vmv.v.i v8, 0
+; CHECK-NEXT:    vse32.v v8, (a0)
+; CHECK-NEXT:    ret
+  store <2 x i32> zeroinitializer, ptr %p
+  ret void
+}
+
+; Not a power of two and requires more than two scalar stores.
+define void @splat_zero_v7i16(ptr %p) {
+; LMULMAX8-RV32-LABEL: splat_zero_v7i16:
+; LMULMAX8-RV32:       # %bb.0:
+; LMULMAX8-RV32-NEXT:    sh zero, 12(a0)
+; LMULMAX8-RV32-NEXT:    vsetivli zero, 4, e16, mf2, ta, ma
+; LMULMAX8-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX8-RV32-NEXT:    vse16.v v8, (a0)
+; LMULMAX8-RV32-NEXT:    vsetivli zero, 8, e16, m1, ta, ma
+; LMULMAX8-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX8-RV32-NEXT:    vsetivli zero, 1, e32, m1, ta, ma
+; LMULMAX8-RV32-NEXT:    vslidedown.vi v8, v8, 2
+; LMULMAX8-RV32-NEXT:    addi a0, a0, 8
+; LMULMAX8-RV32-NEXT:    vse32.v v8, (a0)
+; LMULMAX8-RV32-NEXT:    ret
+;
+; LMULMAX2-RV32-LABEL: splat_zero_v7i16:
+; LMULMAX2-RV32:       # %bb.0:
+; LMULMAX2-RV32-NEXT:    sh zero, 12(a0)
+; LMULMAX2-RV32-NEXT:    vsetivli zero, 4, e16, mf2, ta, ma
+; LMULMAX2-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX2-RV32-NEXT:    vse16.v v8, (a0)
+; LMULMAX2-RV32-NEXT:    vsetivli zero, 8, e16, m1, ta, ma
+; LMULMAX2-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX2-RV32-NEXT:    vsetivli zero, 1, e32, m1, ta, ma
+; LMULMAX2-RV32-NEXT:    vslidedown.vi v8, v8, 2
+; LMULMAX2-RV32-NEXT:    addi a0, a0, 8
+; LMULMAX2-RV32-NEXT:    vse32.v v8, (a0)
+; LMULMAX2-RV32-NEXT:    ret
+;
+; LMULMAX1-RV32-LABEL: splat_zero_v7i16:
+; LMULMAX1-RV32:       # %bb.0:
+; LMULMAX1-RV32-NEXT:    sh zero, 12(a0)
+; LMULMAX1-RV32-NEXT:    vsetivli zero, 4, e16, mf2, ta, ma
+; LMULMAX1-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX1-RV32-NEXT:    vse16.v v8, (a0)
+; LMULMAX1-RV32-NEXT:    vsetivli zero, 8, e16, m1, ta, ma
+; LMULMAX1-RV32-NEXT:    vmv.v.i v8, 0
+; LMULMAX1-RV32-NEXT:    vsetivli zero, 1, e32, m1, ta, ma
+; LMULMAX1-RV32-NEXT:    vslidedown.vi v8, v8, 2
+; LMULMAX1-RV32-NEXT:    addi a0, a0, 8
+; LMULMAX1-RV32-NEXT:    vse32.v v8, (a0)
+; LMULMAX1-RV32-NEXT:    ret
+;
+; LMULMAX8-RV64-LABEL: splat_zero_v7i16:
+; LMULMAX8-RV64:       # %bb.0:
+; LMULMAX8-RV64-NEXT:    sh zero, 12(a0)
+; LMULMAX8-RV64-NEXT:    sw zero, 8(a0)
+; LMULMAX8-RV64-NEXT:    sd zero, 0(a0)
+; LMULMAX8-RV64-NEXT:    ret
+;
+; LMULMAX2-RV64-LABEL: splat_zero_v7i16:
+; LMULMAX2-RV64:       # %bb.0:
+; LMULMAX2-RV64-NEXT:    sh zero, 12(a0)
+; LMULMAX2-RV64-NEXT:    sw zero, 8(a0)
+; LMULMAX2-RV64-NEXT:    sd zero, 0(a0)
+; LMULMAX2-RV64-NEXT:    ret
+;
+; LMULMAX1-RV64-LABEL: splat_zero_v7i16:
+; LMULMAX1-RV64:       # %bb.0:
+; LMULMAX1-RV64-NEXT:    sh zero, 12(a0)
+; LMULMAX1-RV64-NEXT:    sw zero, 8(a0)
+; LMULMAX1-RV64-NEXT:    sd zero, 0(a0)
+; LMULMAX1-RV64-NEXT:    ret
+  store <7 x i16> zeroinitializer, ptr %p
+  ret void
+}
+
 define void @splat_allones_v16i8(ptr %x) {
 ; CHECK-LABEL: splat_allones_v16i8:
 ; CHECK:       # %bb.0:
-- 
2.7.4