; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py ; RUN: llc -mtriple=riscv32 -mattr=+m,+experimental-v -verify-machineinstrs < %s \ ; RUN: | FileCheck %s --check-prefixes=CHECK ; RUN: llc -mtriple=riscv64 -mattr=+m,+experimental-v -verify-machineinstrs < %s \ ; RUN: | FileCheck %s --check-prefixes=CHECK declare @llvm.experimental.stepvector.nxv1i8() define @stepvector_nxv1i8() { ; CHECK-LABEL: stepvector_nxv1i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,mf8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv1i8() ret %v } declare @llvm.experimental.stepvector.nxv2i8() define @stepvector_nxv2i8() { ; CHECK-LABEL: stepvector_nxv2i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,mf4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv2i8() ret %v } declare @llvm.experimental.stepvector.nxv4i8() define @stepvector_nxv4i8() { ; CHECK-LABEL: stepvector_nxv4i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,mf2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv4i8() ret %v } declare @llvm.experimental.stepvector.nxv8i8() define @stepvector_nxv8i8() { ; CHECK-LABEL: stepvector_nxv8i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,m1,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv8i8() ret %v } define @add_stepvector_nxv8i8() { ; CHECK-LABEL: add_stepvector_nxv8i8: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e8,m1,ta,mu ; CHECK-NEXT: vid.v v25 ; CHECK-NEXT: vsll.vi v8, v25, 1 ; CHECK-NEXT: ret entry: %0 = call @llvm.experimental.stepvector.nxv8i8() %1 = call @llvm.experimental.stepvector.nxv8i8() %2 = add %0, %1 ret %2 } define @mul_stepvector_nxv8i8() { ; CHECK-LABEL: mul_stepvector_nxv8i8: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e8,m1,ta,mu ; CHECK-NEXT: vid.v v25 ; CHECK-NEXT: addi a0, zero, 3 ; CHECK-NEXT: vmul.vx v8, v25, a0 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i8 3, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv8i8() %3 = mul %2, %1 ret %3 } define @shl_stepvector_nxv8i8() { ; CHECK-LABEL: shl_stepvector_nxv8i8: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e8,m1,ta,mu ; CHECK-NEXT: vid.v v25 ; CHECK-NEXT: vsll.vi v8, v25, 2 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i8 2, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv8i8() %3 = shl %2, %1 ret %3 } declare @llvm.experimental.stepvector.nxv16i8() define @stepvector_nxv16i8() { ; CHECK-LABEL: stepvector_nxv16i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,m2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv16i8() ret %v } declare @llvm.experimental.stepvector.nxv32i8() define @stepvector_nxv32i8() { ; CHECK-LABEL: stepvector_nxv32i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,m4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv32i8() ret %v } declare @llvm.experimental.stepvector.nxv64i8() define @stepvector_nxv64i8() { ; CHECK-LABEL: stepvector_nxv64i8: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e8,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv64i8() ret %v } declare @llvm.experimental.stepvector.nxv1i16() define @stepvector_nxv1i16() { ; CHECK-LABEL: stepvector_nxv1i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,mf4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv1i16() ret %v } declare @llvm.experimental.stepvector.nxv2i16() define @stepvector_nxv2i16() { ; CHECK-LABEL: stepvector_nxv2i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,mf2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv2i16() ret %v } declare @llvm.experimental.stepvector.nxv4i16() define @stepvector_nxv4i16() { ; CHECK-LABEL: stepvector_nxv4i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,m1,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv4i16() ret %v } declare @llvm.experimental.stepvector.nxv8i16() define @stepvector_nxv8i16() { ; CHECK-LABEL: stepvector_nxv8i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,m2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv8i16() ret %v } declare @llvm.experimental.stepvector.nxv16i16() define @stepvector_nxv16i16() { ; CHECK-LABEL: stepvector_nxv16i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,m4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv16i16() ret %v } define @add_stepvector_nxv16i16() { ; CHECK-LABEL: add_stepvector_nxv16i16: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e16,m4,ta,mu ; CHECK-NEXT: vid.v v28 ; CHECK-NEXT: vsll.vi v8, v28, 1 ; CHECK-NEXT: ret entry: %0 = call @llvm.experimental.stepvector.nxv16i16() %1 = call @llvm.experimental.stepvector.nxv16i16() %2 = add %0, %1 ret %2 } define @mul_stepvector_nxv16i16() { ; CHECK-LABEL: mul_stepvector_nxv16i16: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e16,m4,ta,mu ; CHECK-NEXT: vid.v v28 ; CHECK-NEXT: addi a0, zero, 3 ; CHECK-NEXT: vmul.vx v8, v28, a0 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i16 3, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i16() %3 = mul %2, %1 ret %3 } define @shl_stepvector_nxv16i16() { ; CHECK-LABEL: shl_stepvector_nxv16i16: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e16,m4,ta,mu ; CHECK-NEXT: vid.v v28 ; CHECK-NEXT: vsll.vi v8, v28, 2 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i16 2, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i16() %3 = shl %2, %1 ret %3 } declare @llvm.experimental.stepvector.nxv32i16() define @stepvector_nxv32i16() { ; CHECK-LABEL: stepvector_nxv32i16: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e16,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv32i16() ret %v } declare @llvm.experimental.stepvector.nxv1i32() define @stepvector_nxv1i32() { ; CHECK-LABEL: stepvector_nxv1i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32,mf2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv1i32() ret %v } declare @llvm.experimental.stepvector.nxv2i32() define @stepvector_nxv2i32() { ; CHECK-LABEL: stepvector_nxv2i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32,m1,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv2i32() ret %v } declare @llvm.experimental.stepvector.nxv4i32() define @stepvector_nxv4i32() { ; CHECK-LABEL: stepvector_nxv4i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32,m2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv4i32() ret %v } declare @llvm.experimental.stepvector.nxv8i32() define @stepvector_nxv8i32() { ; CHECK-LABEL: stepvector_nxv8i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32,m4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv8i32() ret %v } declare @llvm.experimental.stepvector.nxv16i32() define @stepvector_nxv16i32() { ; CHECK-LABEL: stepvector_nxv16i32: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e32,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv16i32() ret %v } define @add_stepvector_nxv16i32() { ; CHECK-LABEL: add_stepvector_nxv16i32: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e32,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: vsll.vi v8, v8, 1 ; CHECK-NEXT: ret entry: %0 = call @llvm.experimental.stepvector.nxv16i32() %1 = call @llvm.experimental.stepvector.nxv16i32() %2 = add %0, %1 ret %2 } define @mul_stepvector_nxv16i32() { ; CHECK-LABEL: mul_stepvector_nxv16i32: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e32,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: addi a0, zero, 3 ; CHECK-NEXT: vmul.vx v8, v8, a0 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i32 3, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i32() %3 = mul %2, %1 ret %3 } define @shl_stepvector_nxv16i32() { ; CHECK-LABEL: shl_stepvector_nxv16i32: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e32,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: vsll.vi v8, v8, 2 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i32 2, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i32() %3 = shl %2, %1 ret %3 } declare @llvm.experimental.stepvector.nxv1i64() define @stepvector_nxv1i64() { ; CHECK-LABEL: stepvector_nxv1i64: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e64,m1,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv1i64() ret %v } declare @llvm.experimental.stepvector.nxv2i64() define @stepvector_nxv2i64() { ; CHECK-LABEL: stepvector_nxv2i64: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e64,m2,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv2i64() ret %v } declare @llvm.experimental.stepvector.nxv4i64() define @stepvector_nxv4i64() { ; CHECK-LABEL: stepvector_nxv4i64: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e64,m4,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv4i64() ret %v } declare @llvm.experimental.stepvector.nxv8i64() define @stepvector_nxv8i64() { ; CHECK-LABEL: stepvector_nxv8i64: ; CHECK: # %bb.0: ; CHECK-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv8i64() ret %v } define @add_stepvector_nxv8i64() { ; CHECK-LABEL: add_stepvector_nxv8i64: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: vsll.vi v8, v8, 1 ; CHECK-NEXT: ret entry: %0 = call @llvm.experimental.stepvector.nxv8i64() %1 = call @llvm.experimental.stepvector.nxv8i64() %2 = add %0, %1 ret %2 } define @mul_stepvector_nxv8i64() { ; CHECK-LABEL: mul_stepvector_nxv8i64: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: addi a0, zero, 3 ; CHECK-NEXT: vmul.vx v8, v8, a0 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i64 3, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv8i64() %3 = mul %2, %1 ret %3 } define @shl_stepvector_nxv8i64() { ; CHECK-LABEL: shl_stepvector_nxv8i64: ; CHECK: # %bb.0: # %entry ; CHECK-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: vsll.vi v8, v8, 2 ; CHECK-NEXT: ret entry: %0 = insertelement poison, i64 2, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv8i64() %3 = shl %2, %1 ret %3 } declare @llvm.experimental.stepvector.nxv16i64() define @stepvector_nxv16i64() { ; CHECK-LABEL: stepvector_nxv16i64: ; CHECK: # %bb.0: ; CHECK-NEXT: csrr a0, vlenb ; CHECK-NEXT: vsetvli a1, zero, e64,m8,ta,mu ; CHECK-NEXT: vid.v v8 ; CHECK-NEXT: vadd.vx v16, v8, a0 ; CHECK-NEXT: ret %v = call @llvm.experimental.stepvector.nxv16i64() ret %v } define @add_stepvector_nxv16i64() { ; RV32-LABEL: add_stepvector_nxv16i64: ; RV32: # %bb.0: # %entry ; RV32-NEXT: csrr a0, vlenb ; RV32-NEXT: slli a0, a0, 1 ; RV32-NEXT: vsetvli a1, zero, e64,m8,ta,mu ; RV32-NEXT: vmv.v.x v8, a0 ; RV32-NEXT: addi a0, zero, 32 ; RV32-NEXT: vsll.vx v8, v8, a0 ; RV32-NEXT: vsrl.vx v16, v8, a0 ; RV32-NEXT: vid.v v8 ; RV32-NEXT: vsll.vi v8, v8, 1 ; RV32-NEXT: vadd.vv v16, v8, v16 ; RV32-NEXT: ret ; ; RV64-LABEL: add_stepvector_nxv16i64: ; RV64: # %bb.0: # %entry ; RV64-NEXT: csrr a0, vlenb ; RV64-NEXT: slli a0, a0, 1 ; RV64-NEXT: vsetvli a1, zero, e64,m8,ta,mu ; RV64-NEXT: vid.v v8 ; RV64-NEXT: vsll.vi v8, v8, 1 ; RV64-NEXT: vadd.vx v16, v8, a0 ; RV64-NEXT: ret entry: %0 = call @llvm.experimental.stepvector.nxv16i64() %1 = call @llvm.experimental.stepvector.nxv16i64() %2 = add %0, %1 ret %2 } define @mul_stepvector_nxv16i64() { ; RV32-LABEL: mul_stepvector_nxv16i64: ; RV32: # %bb.0: # %entry ; RV32-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; RV32-NEXT: vid.v v8 ; RV32-NEXT: addi a0, zero, 3 ; RV32-NEXT: vmul.vx v8, v8, a0 ; RV32-NEXT: csrr a0, vlenb ; RV32-NEXT: srli a0, a0, 3 ; RV32-NEXT: addi a1, zero, 24 ; RV32-NEXT: mul a0, a0, a1 ; RV32-NEXT: vmv.v.x v16, a0 ; RV32-NEXT: addi a0, zero, 32 ; RV32-NEXT: vsll.vx v16, v16, a0 ; RV32-NEXT: vsrl.vx v16, v16, a0 ; RV32-NEXT: vadd.vv v16, v8, v16 ; RV32-NEXT: ret ; ; RV64-LABEL: mul_stepvector_nxv16i64: ; RV64: # %bb.0: # %entry ; RV64-NEXT: vsetvli a0, zero, e64,m8,ta,mu ; RV64-NEXT: vid.v v8 ; RV64-NEXT: addi a0, zero, 3 ; RV64-NEXT: vmul.vx v8, v8, a0 ; RV64-NEXT: csrr a0, vlenb ; RV64-NEXT: srli a0, a0, 3 ; RV64-NEXT: addi a1, zero, 24 ; RV64-NEXT: mul a0, a0, a1 ; RV64-NEXT: vadd.vx v16, v8, a0 ; RV64-NEXT: ret entry: %0 = insertelement poison, i64 3, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i64() %3 = mul %2, %1 ret %3 } define @shl_stepvector_nxv16i64() { ; RV32-LABEL: shl_stepvector_nxv16i64: ; RV32: # %bb.0: # %entry ; RV32-NEXT: csrr a0, vlenb ; RV32-NEXT: slli a0, a0, 2 ; RV32-NEXT: vsetvli a1, zero, e64,m8,ta,mu ; RV32-NEXT: vmv.v.x v8, a0 ; RV32-NEXT: addi a0, zero, 32 ; RV32-NEXT: vsll.vx v8, v8, a0 ; RV32-NEXT: vsrl.vx v16, v8, a0 ; RV32-NEXT: vid.v v8 ; RV32-NEXT: vsll.vi v8, v8, 2 ; RV32-NEXT: vadd.vv v16, v8, v16 ; RV32-NEXT: ret ; ; RV64-LABEL: shl_stepvector_nxv16i64: ; RV64: # %bb.0: # %entry ; RV64-NEXT: csrr a0, vlenb ; RV64-NEXT: slli a0, a0, 2 ; RV64-NEXT: vsetvli a1, zero, e64,m8,ta,mu ; RV64-NEXT: vid.v v8 ; RV64-NEXT: vsll.vi v8, v8, 2 ; RV64-NEXT: vadd.vx v16, v8, a0 ; RV64-NEXT: ret entry: %0 = insertelement poison, i64 2, i32 0 %1 = shufflevector %0, poison, zeroinitializer %2 = call @llvm.experimental.stepvector.nxv16i64() %3 = shl %2, %1 ret %3 }