diff --git a/llvm/test/CodeGen/RISCV/rvv/concat-vector-insert-elt.ll b/llvm/test/CodeGen/RISCV/rvv/concat-vector-insert-elt.ll new file mode 100644 index 0000000000000..9193f7aef4b87 --- /dev/null +++ b/llvm/test/CodeGen/RISCV/rvv/concat-vector-insert-elt.ll @@ -0,0 +1,241 @@ +; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py +; RUN: llc -mtriple=riscv32 -mattr=+v -target-abi=ilp32 \ +; RUN: -verify-machineinstrs < %s | FileCheck %s --check-prefixes=CHECK,RV32 +; RUN: llc -mtriple=riscv64 -mattr=+v -target-abi=lp64 \ +; RUN: -verify-machineinstrs < %s | FileCheck %s --check-prefixes=CHECK,RV64 + +define void @v4xi8_concat_vector_insert_idx0(ptr %a, ptr %b, i8 %x) { +; CHECK-LABEL: v4xi8_concat_vector_insert_idx0: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetivli zero, 2, e8, mf8, ta, ma +; CHECK-NEXT: vle8.v v8, (a0) +; CHECK-NEXT: vle8.v v9, (a1) +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vslideup.vi v8, v9, 2 +; CHECK-NEXT: vmv.s.x v9, a2 +; CHECK-NEXT: vsetivli zero, 2, e8, mf4, tu, ma +; CHECK-NEXT: vslideup.vi v8, v9, 1 +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vse8.v v8, (a0) +; CHECK-NEXT: ret + %v1 = load <2 x i8>, ptr %a + %v2 = load <2 x i8>, ptr %b + %concat = shufflevector <2 x i8> %v1, <2 x i8> %v2, <4 x i32> + %ins = insertelement <4 x i8> %concat, i8 %x, i32 1 + store <4 x i8> %ins, ptr %a + ret void +} + +define void @v4xi8_concat_vector_insert_idx1(ptr %a, ptr %b, i8 %x) { +; CHECK-LABEL: v4xi8_concat_vector_insert_idx1: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetivli zero, 2, e8, mf8, ta, ma +; CHECK-NEXT: vle8.v v8, (a0) +; CHECK-NEXT: vle8.v v9, (a1) +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vslideup.vi v8, v9, 2 +; CHECK-NEXT: vmv.s.x v9, a2 +; CHECK-NEXT: vsetivli zero, 2, e8, mf4, tu, ma +; CHECK-NEXT: vslideup.vi v8, v9, 1 +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vse8.v v8, (a0) +; CHECK-NEXT: ret + %v1 = load <2 x i8>, ptr %a + %v2 = load <2 x i8>, ptr %b + %concat = shufflevector <2 x i8> %v1, <2 x i8> %v2, <4 x i32> + %ins = insertelement <4 x i8> %concat, i8 %x, i32 1 + store <4 x i8> %ins, ptr %a + ret void +} + +define void @v4xi8_concat_vector_insert_idx2(ptr %a, ptr %b, i8 %x) { +; CHECK-LABEL: v4xi8_concat_vector_insert_idx2: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetivli zero, 2, e8, mf8, ta, ma +; CHECK-NEXT: vle8.v v8, (a0) +; CHECK-NEXT: vle8.v v9, (a1) +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vslideup.vi v8, v9, 2 +; CHECK-NEXT: vmv.s.x v9, a2 +; CHECK-NEXT: vsetivli zero, 3, e8, mf4, tu, ma +; CHECK-NEXT: vslideup.vi v8, v9, 2 +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vse8.v v8, (a0) +; CHECK-NEXT: ret + %v1 = load <2 x i8>, ptr %a + %v2 = load <2 x i8>, ptr %b + %concat = shufflevector <2 x i8> %v1, <2 x i8> %v2, <4 x i32> + %ins = insertelement <4 x i8> %concat, i8 %x, i32 2 + store <4 x i8> %ins, ptr %a + ret void +} + +define void @v4xi8_concat_vector_insert_idx3(ptr %a, ptr %b, i8 %x) { +; CHECK-LABEL: v4xi8_concat_vector_insert_idx3: +; CHECK: # %bb.0: +; CHECK-NEXT: vsetivli zero, 2, e8, mf8, ta, ma +; CHECK-NEXT: vle8.v v8, (a0) +; CHECK-NEXT: vle8.v v9, (a1) +; CHECK-NEXT: vsetivli zero, 4, e8, mf4, ta, ma +; CHECK-NEXT: vslideup.vi v8, v9, 2 +; CHECK-NEXT: vmv.s.x v9, a2 +; CHECK-NEXT: vslideup.vi v8, v9, 3 +; CHECK-NEXT: vse8.v v8, (a0) +; CHECK-NEXT: ret + %v1 = load <2 x i8>, ptr %a + %v2 = load <2 x i8>, ptr %b + %concat = shufflevector <2 x i8> %v1, <2 x i8> %v2, <4 x i32> + %ins = insertelement <4 x i8> %concat, i8 %x, i32 3 + store <4 x i8> %ins, ptr %a + ret void +} + +define void @v4xi64_concat_vector_insert_idx0(ptr %a, ptr %b, i64 %x) { +; RV32-LABEL: v4xi64_concat_vector_insert_idx0: +; RV32: # %bb.0: +; RV32-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV32-NEXT: vle64.v v8, (a0) +; RV32-NEXT: vle64.v v10, (a1) +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vslideup.vi v8, v10, 2 +; RV32-NEXT: vsetivli zero, 2, e32, m1, ta, ma +; RV32-NEXT: vslide1down.vx v10, v8, a2 +; RV32-NEXT: vslide1down.vx v10, v10, a3 +; RV32-NEXT: vsetivli zero, 2, e64, m1, tu, ma +; RV32-NEXT: vslideup.vi v8, v10, 1 +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vse64.v v8, (a0) +; RV32-NEXT: ret +; +; RV64-LABEL: v4xi64_concat_vector_insert_idx0: +; RV64: # %bb.0: +; RV64-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV64-NEXT: vle64.v v8, (a0) +; RV64-NEXT: vle64.v v10, (a1) +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vslideup.vi v8, v10, 2 +; RV64-NEXT: vmv.s.x v10, a2 +; RV64-NEXT: vsetivli zero, 2, e64, m1, tu, ma +; RV64-NEXT: vslideup.vi v8, v10, 1 +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vse64.v v8, (a0) +; RV64-NEXT: ret + %v1 = load <2 x i64>, ptr %a + %v2 = load <2 x i64>, ptr %b + %concat = shufflevector <2 x i64> %v1, <2 x i64> %v2, <4 x i32> + %ins = insertelement <4 x i64> %concat, i64 %x, i32 1 + store <4 x i64> %ins, ptr %a + ret void +} + +define void @v4xi64_concat_vector_insert_idx1(ptr %a, ptr %b, i64 %x) { +; RV32-LABEL: v4xi64_concat_vector_insert_idx1: +; RV32: # %bb.0: +; RV32-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV32-NEXT: vle64.v v8, (a0) +; RV32-NEXT: vle64.v v10, (a1) +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vslideup.vi v8, v10, 2 +; RV32-NEXT: vsetivli zero, 2, e32, m1, ta, ma +; RV32-NEXT: vslide1down.vx v10, v8, a2 +; RV32-NEXT: vslide1down.vx v10, v10, a3 +; RV32-NEXT: vsetivli zero, 2, e64, m1, tu, ma +; RV32-NEXT: vslideup.vi v8, v10, 1 +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vse64.v v8, (a0) +; RV32-NEXT: ret +; +; RV64-LABEL: v4xi64_concat_vector_insert_idx1: +; RV64: # %bb.0: +; RV64-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV64-NEXT: vle64.v v8, (a0) +; RV64-NEXT: vle64.v v10, (a1) +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vslideup.vi v8, v10, 2 +; RV64-NEXT: vmv.s.x v10, a2 +; RV64-NEXT: vsetivli zero, 2, e64, m1, tu, ma +; RV64-NEXT: vslideup.vi v8, v10, 1 +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vse64.v v8, (a0) +; RV64-NEXT: ret + %v1 = load <2 x i64>, ptr %a + %v2 = load <2 x i64>, ptr %b + %concat = shufflevector <2 x i64> %v1, <2 x i64> %v2, <4 x i32> + %ins = insertelement <4 x i64> %concat, i64 %x, i32 1 + store <4 x i64> %ins, ptr %a + ret void +} + +define void @v4xi64_concat_vector_insert_idx2(ptr %a, ptr %b, i64 %x) { +; RV32-LABEL: v4xi64_concat_vector_insert_idx2: +; RV32: # %bb.0: +; RV32-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV32-NEXT: vle64.v v8, (a0) +; RV32-NEXT: vle64.v v10, (a1) +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vslideup.vi v8, v10, 2 +; RV32-NEXT: vsetivli zero, 2, e32, m2, ta, ma +; RV32-NEXT: vslide1down.vx v10, v8, a2 +; RV32-NEXT: vslide1down.vx v10, v10, a3 +; RV32-NEXT: vsetivli zero, 3, e64, m2, tu, ma +; RV32-NEXT: vslideup.vi v8, v10, 2 +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vse64.v v8, (a0) +; RV32-NEXT: ret +; +; RV64-LABEL: v4xi64_concat_vector_insert_idx2: +; RV64: # %bb.0: +; RV64-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV64-NEXT: vle64.v v8, (a0) +; RV64-NEXT: vle64.v v10, (a1) +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vslideup.vi v8, v10, 2 +; RV64-NEXT: vmv.s.x v10, a2 +; RV64-NEXT: vsetivli zero, 3, e64, m2, tu, ma +; RV64-NEXT: vslideup.vi v8, v10, 2 +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vse64.v v8, (a0) +; RV64-NEXT: ret + %v1 = load <2 x i64>, ptr %a + %v2 = load <2 x i64>, ptr %b + %concat = shufflevector <2 x i64> %v1, <2 x i64> %v2, <4 x i32> + %ins = insertelement <4 x i64> %concat, i64 %x, i32 2 + store <4 x i64> %ins, ptr %a + ret void +} + +define void @v4xi64_concat_vector_insert_idx3(ptr %a, ptr %b, i64 %x) { +; RV32-LABEL: v4xi64_concat_vector_insert_idx3: +; RV32: # %bb.0: +; RV32-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV32-NEXT: vle64.v v8, (a0) +; RV32-NEXT: vle64.v v10, (a1) +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vslideup.vi v8, v10, 2 +; RV32-NEXT: vsetivli zero, 2, e32, m2, ta, ma +; RV32-NEXT: vslide1down.vx v10, v8, a2 +; RV32-NEXT: vslide1down.vx v10, v10, a3 +; RV32-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV32-NEXT: vslideup.vi v8, v10, 3 +; RV32-NEXT: vse64.v v8, (a0) +; RV32-NEXT: ret +; +; RV64-LABEL: v4xi64_concat_vector_insert_idx3: +; RV64: # %bb.0: +; RV64-NEXT: vsetivli zero, 2, e64, m1, ta, ma +; RV64-NEXT: vle64.v v8, (a0) +; RV64-NEXT: vle64.v v10, (a1) +; RV64-NEXT: vsetivli zero, 4, e64, m2, ta, ma +; RV64-NEXT: vslideup.vi v8, v10, 2 +; RV64-NEXT: vmv.s.x v10, a2 +; RV64-NEXT: vslideup.vi v8, v10, 3 +; RV64-NEXT: vse64.v v8, (a0) +; RV64-NEXT: ret + %v1 = load <2 x i64>, ptr %a + %v2 = load <2 x i64>, ptr %b + %concat = shufflevector <2 x i64> %v1, <2 x i64> %v2, <4 x i32> + %ins = insertelement <4 x i64> %concat, i64 %x, i32 3 + store <4 x i64> %ins, ptr %a + ret void +} + diff --git a/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-buildvec-of-binop.ll b/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-buildvec-of-binop.ll index 8055944fc5468..2ffca983ac102 100644 --- a/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-buildvec-of-binop.ll +++ b/llvm/test/CodeGen/RISCV/rvv/fixed-vectors-buildvec-of-binop.ll @@ -566,3 +566,41 @@ define <8 x i32> @add_constant_rhs_8xi32_vector_in3(<8 x i32> %vin, i32 %a, i32 %v3 = insertelement <8 x i32> %v2, i32 %e3, i32 6 ret <8 x i32> %v3 } + +define <8 x i32> @add_constant_rhs_8xi32_partial(<8 x i32> %vin, i32 %a, i32 %b, i32 %c, i32 %d) { +; CHECK-LABEL: add_constant_rhs_8xi32_partial: +; CHECK: # %bb.0: +; CHECK-NEXT: lui a4, %hi(.LCPI19_0) +; CHECK-NEXT: addi a4, a4, %lo(.LCPI19_0) +; CHECK-NEXT: vsetivli zero, 8, e32, m2, ta, ma +; CHECK-NEXT: vle32.v v10, (a4) +; CHECK-NEXT: vadd.vv v8, v8, v10 +; CHECK-NEXT: addi a0, a0, 23 +; CHECK-NEXT: addi a1, a1, 25 +; CHECK-NEXT: addi a2, a2, 1 +; CHECK-NEXT: addi a3, a3, 2047 +; CHECK-NEXT: addi a3, a3, 308 +; CHECK-NEXT: vmv.s.x v10, a0 +; CHECK-NEXT: vsetivli zero, 5, e32, m2, tu, ma +; CHECK-NEXT: vslideup.vi v8, v10, 4 +; CHECK-NEXT: vmv.s.x v10, a1 +; CHECK-NEXT: vsetivli zero, 6, e32, m2, tu, ma +; CHECK-NEXT: vslideup.vi v8, v10, 5 +; CHECK-NEXT: vmv.s.x v10, a2 +; CHECK-NEXT: vsetivli zero, 7, e32, m2, tu, ma +; CHECK-NEXT: vslideup.vi v8, v10, 6 +; CHECK-NEXT: vmv.s.x v10, a3 +; CHECK-NEXT: vsetivli zero, 8, e32, m2, ta, ma +; CHECK-NEXT: vslideup.vi v8, v10, 7 +; CHECK-NEXT: ret + %vadd = add <8 x i32> %vin, + %e0 = add i32 %a, 23 + %e1 = add i32 %b, 25 + %e2 = add i32 %c, 1 + %e3 = add i32 %d, 2355 + %v0 = insertelement <8 x i32> %vadd, i32 %e0, i32 4 + %v1 = insertelement <8 x i32> %v0, i32 %e1, i32 5 + %v2 = insertelement <8 x i32> %v1, i32 %e2, i32 6 + %v3 = insertelement <8 x i32> %v2, i32 %e3, i32 7 + ret <8 x i32> %v3 +}