[RISCV][test] Add test of binop followed by extractelement.

Reviewed By: reames Differential Revision: https://reviews.llvm.org/D129544
llvm · Jul 13, 2022 · 9049c46 · 9049c46
1 parent 2b04532
commit 9049c46
Show file tree

Hide file tree

Showing 5 changed files with 499 additions and 0 deletions.
diff --git a/llvm/test/CodeGen/RISCV/rvv/extractelt-fp-rv32.ll b/llvm/test/CodeGen/RISCV/rvv/extractelt-fp-rv32.ll
@@ -481,3 +481,73 @@ define double @extractelt_nxv8f64_idx(<vscale x 8 x double> %v, i32 %idx) {
   %r = extractelement <vscale x 8 x double> %v, i32 %idx
   ret double %r
 }
+
+define float @extractelt_fadd_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fadd_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI45_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI45_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfadd.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 2
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fadd <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 2
+  ret float %ext
+}
+
+define float @extractelt_fsub_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fsub_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI46_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI46_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfrsub.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 1
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fsub <vscale x 4 x float> %splat, %x
+  %ext = extractelement <vscale x 4 x float> %bo, i32 1
+  ret float %ext
+}
+
+define float @extractelt_fmul_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fmul_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI47_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI47_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfmul.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 3
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fmul <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 3
+  ret float %ext
+}
+
+define float @extractelt_fdiv_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fdiv_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI48_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI48_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfdiv.vf v8, v8, ft0
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fdiv <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 0
+  ret float %ext
+}
diff --git a/llvm/test/CodeGen/RISCV/rvv/extractelt-fp-rv64.ll b/llvm/test/CodeGen/RISCV/rvv/extractelt-fp-rv64.ll
@@ -510,3 +510,73 @@ define void @store_vfmv_f_s_nxv8f64(<vscale x 8 x double>* %x, double* %p) {
 }
 
 declare double @llvm.riscv.vfmv.f.s.nxv8f64(<vscale x 8 x double>)
+
+define float @extractelt_fadd_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fadd_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI47_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI47_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfadd.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 2
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fadd <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 2
+  ret float %ext
+}
+
+define float @extractelt_fsub_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fsub_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI48_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI48_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfrsub.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 1
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fsub <vscale x 4 x float> %splat, %x
+  %ext = extractelement <vscale x 4 x float> %bo, i32 1
+  ret float %ext
+}
+
+define float @extractelt_fmul_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fmul_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI49_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI49_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfmul.vf v8, v8, ft0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 3
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fmul <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 3
+  ret float %ext
+}
+
+define float @extractelt_fdiv_nxv4f32_splat(<vscale x 4 x float> %x) {
+; CHECK-LABEL: extractelt_fdiv_nxv4f32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, %hi(.LCPI50_0)
+; CHECK-NEXT:    flw ft0, %lo(.LCPI50_0)(a0)
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vfdiv.vf v8, v8, ft0
+; CHECK-NEXT:    vfmv.f.s fa0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x float> poison, float 3.0, i32 0
+  %splat = shufflevector <vscale x 4 x float> %head, <vscale x 4 x float> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = fdiv <vscale x 4 x float> %x, %splat
+  %ext = extractelement <vscale x 4 x float> %bo, i32 0
+  ret float %ext
+}
diff --git a/llvm/test/CodeGen/RISCV/rvv/extractelt-int-rv32.ll b/llvm/test/CodeGen/RISCV/rvv/extractelt-int-rv32.ll
@@ -741,3 +741,88 @@ define i64 @extractelt_nxv8i64_idx(<vscale x 8 x i64> %v, i32 %idx) {
   %r = extractelement <vscale x 8 x i64> %v, i32 %idx
   ret i64 %r
 }
+
+define i32 @extractelt_add_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_add_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vadd.vi v8, v8, 3
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 2
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = add <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 2
+  ret i32 %ext
+}
+
+define i32 @extractelt_sub_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_sub_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vrsub.vi v8, v8, 3
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 1
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sub <vscale x 4 x i32> %splat, %x
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 1
+  ret i32 %ext
+}
+
+define i32 @extractelt_mul_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_mul_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    li a0, 3
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmul.vx v8, v8, a0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 3
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = mul <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 3
+  ret i32 %ext
+}
+
+define i32 @extractelt_sdiv_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_sdiv_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, 349525
+; CHECK-NEXT:    addi a0, a0, 1366
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmulh.vx v8, v8, a0
+; CHECK-NEXT:    vsrl.vi v10, v8, 31
+; CHECK-NEXT:    vadd.vv v8, v8, v10
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sdiv <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 0
+  ret i32 %ext
+}
+
+define i32 @extractelt_udiv_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_udiv_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, 349525
+; CHECK-NEXT:    addi a0, a0, 1366
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmulh.vx v8, v8, a0
+; CHECK-NEXT:    vsrl.vi v10, v8, 31
+; CHECK-NEXT:    vadd.vv v8, v8, v10
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sdiv <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 0
+  ret i32 %ext
+}
diff --git a/llvm/test/CodeGen/RISCV/rvv/extractelt-int-rv64.ll b/llvm/test/CodeGen/RISCV/rvv/extractelt-int-rv64.ll
@@ -705,3 +705,88 @@ define i64 @extractelt_nxv8i64_idx(<vscale x 8 x i64> %v, i32 signext %idx) {
   %r = extractelement <vscale x 8 x i64> %v, i32 %idx
   ret i64 %r
 }
+
+define i32 @extractelt_add_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_add_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vadd.vi v8, v8, 3
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 2
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = add <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 2
+  ret i32 %ext
+}
+
+define i32 @extractelt_sub_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_sub_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    vsetvli a0, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vrsub.vi v8, v8, 3
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 1
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sub <vscale x 4 x i32> %splat, %x
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 1
+  ret i32 %ext
+}
+
+define i32 @extractelt_mul_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_mul_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    li a0, 3
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmul.vx v8, v8, a0
+; CHECK-NEXT:    vsetivli zero, 1, e32, m2, ta, mu
+; CHECK-NEXT:    vslidedown.vi v8, v8, 3
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = mul <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 3
+  ret i32 %ext
+}
+
+define i32 @extractelt_sdiv_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_sdiv_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, 349525
+; CHECK-NEXT:    addiw a0, a0, 1366
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmulh.vx v8, v8, a0
+; CHECK-NEXT:    vsrl.vi v10, v8, 31
+; CHECK-NEXT:    vadd.vv v8, v8, v10
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sdiv <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 0
+  ret i32 %ext
+}
+
+define i32 @extractelt_udiv_nxv4i32_splat(<vscale x 4 x i32> %x) {
+; CHECK-LABEL: extractelt_udiv_nxv4i32_splat:
+; CHECK:       # %bb.0:
+; CHECK-NEXT:    lui a0, 349525
+; CHECK-NEXT:    addiw a0, a0, 1366
+; CHECK-NEXT:    vsetvli a1, zero, e32, m2, ta, mu
+; CHECK-NEXT:    vmulh.vx v8, v8, a0
+; CHECK-NEXT:    vsrl.vi v10, v8, 31
+; CHECK-NEXT:    vadd.vv v8, v8, v10
+; CHECK-NEXT:    vmv.x.s a0, v8
+; CHECK-NEXT:    ret
+  %head = insertelement <vscale x 4 x i32> poison, i32 3, i32 0
+  %splat = shufflevector <vscale x 4 x i32> %head, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+  %bo = sdiv <vscale x 4 x i32> %x, %splat
+  %ext = extractelement <vscale x 4 x i32> %bo, i32 0
+  ret i32 %ext
+}