|
| 1 | +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 3 |
| 2 | +; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv32 -mattr=+m,+v | FileCheck %s |
| 3 | +; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv64 -mattr=+m,+v | FileCheck %s |
| 4 | +; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv32 -mattr=+v,+experimental-zvbb | FileCheck %s |
| 5 | +; RUN: opt < %s -passes=slp-vectorizer -S -mtriple=riscv64 -mattr=+v,+experimental-zvbb | FileCheck %s |
| 6 | + |
| 7 | +define <4 x float> @rint_v4f32(ptr %a) { |
| 8 | +; CHECK-LABEL: define <4 x float> @rint_v4f32( |
| 9 | +; CHECK-SAME: ptr [[A:%.*]]) #[[ATTR0:[0-9]+]] { |
| 10 | +; CHECK-NEXT: entry: |
| 11 | +; CHECK-NEXT: [[TMP0:%.*]] = load <4 x float>, ptr [[A]], align 16 |
| 12 | +; CHECK-NEXT: [[TMP1:%.*]] = call <4 x float> @llvm.rint.v4f32(<4 x float> [[TMP0]]) |
| 13 | +; CHECK-NEXT: ret <4 x float> [[TMP1]] |
| 14 | +; |
| 15 | +entry: |
| 16 | + %0 = load <4 x float>, ptr %a |
| 17 | + %vecext = extractelement <4 x float> %0, i64 0 |
| 18 | + %1 = call float @llvm.rint.f32(float %vecext) |
| 19 | + %vecins = insertelement <4 x float> undef, float %1, i64 0 |
| 20 | + %vecext.1 = extractelement <4 x float> %0, i64 1 |
| 21 | + %2 = call float @llvm.rint.f32(float %vecext.1) |
| 22 | + %vecins.1 = insertelement <4 x float> %vecins, float %2, i64 1 |
| 23 | + %vecext.2 = extractelement <4 x float> %0, i64 2 |
| 24 | + %3 = call float @llvm.rint.f32(float %vecext.2) |
| 25 | + %vecins.2 = insertelement <4 x float> %vecins.1, float %3, i64 2 |
| 26 | + %vecext.3 = extractelement <4 x float> %0, i64 3 |
| 27 | + %4 = call float @llvm.rint.f32(float %vecext.3) |
| 28 | + %vecins.3 = insertelement <4 x float> %vecins.2, float %4, i64 3 |
| 29 | + ret <4 x float> %vecins.3 |
| 30 | +} |
| 31 | + |
| 32 | +define <4 x i64> @lrint_v4i64f32(ptr %a) { |
| 33 | +; CHECK-LABEL: define <4 x i64> @lrint_v4i64f32( |
| 34 | +; CHECK-SAME: ptr [[A:%.*]]) #[[ATTR0]] { |
| 35 | +; CHECK-NEXT: entry: |
| 36 | +; CHECK-NEXT: [[TMP0:%.*]] = load <4 x float>, ptr [[A]], align 16 |
| 37 | +; CHECK-NEXT: [[VECEXT:%.*]] = extractelement <4 x float> [[TMP0]], i64 0 |
| 38 | +; CHECK-NEXT: [[TMP1:%.*]] = call i64 @llvm.lrint.i64.f32(float [[VECEXT]]) |
| 39 | +; CHECK-NEXT: [[VECINS:%.*]] = insertelement <4 x i64> undef, i64 [[TMP1]], i64 0 |
| 40 | +; CHECK-NEXT: [[VECEXT_1:%.*]] = extractelement <4 x float> [[TMP0]], i64 1 |
| 41 | +; CHECK-NEXT: [[TMP2:%.*]] = call i64 @llvm.lrint.i64.f32(float [[VECEXT_1]]) |
| 42 | +; CHECK-NEXT: [[VECINS_1:%.*]] = insertelement <4 x i64> [[VECINS]], i64 [[TMP2]], i64 1 |
| 43 | +; CHECK-NEXT: [[VECEXT_2:%.*]] = extractelement <4 x float> [[TMP0]], i64 2 |
| 44 | +; CHECK-NEXT: [[TMP3:%.*]] = call i64 @llvm.lrint.i64.f32(float [[VECEXT_2]]) |
| 45 | +; CHECK-NEXT: [[VECINS_2:%.*]] = insertelement <4 x i64> [[VECINS_1]], i64 [[TMP3]], i64 2 |
| 46 | +; CHECK-NEXT: [[VECEXT_3:%.*]] = extractelement <4 x float> [[TMP0]], i64 3 |
| 47 | +; CHECK-NEXT: [[TMP4:%.*]] = call i64 @llvm.lrint.i64.f32(float [[VECEXT_3]]) |
| 48 | +; CHECK-NEXT: [[VECINS_3:%.*]] = insertelement <4 x i64> [[VECINS_2]], i64 [[TMP4]], i64 3 |
| 49 | +; CHECK-NEXT: ret <4 x i64> [[VECINS_3]] |
| 50 | +; |
| 51 | +entry: |
| 52 | + %0 = load <4 x float>, ptr %a |
| 53 | + %vecext = extractelement <4 x float> %0, i64 0 |
| 54 | + %1 = call i64 @llvm.lrint.i64.f32(float %vecext) |
| 55 | + %vecins = insertelement <4 x i64> undef, i64 %1, i64 0 |
| 56 | + %vecext.1 = extractelement <4 x float> %0, i64 1 |
| 57 | + %2 = call i64 @llvm.lrint.i64.f32(float %vecext.1) |
| 58 | + %vecins.1 = insertelement <4 x i64> %vecins, i64 %2, i64 1 |
| 59 | + %vecext.2 = extractelement <4 x float> %0, i64 2 |
| 60 | + %3 = call i64 @llvm.lrint.i64.f32(float %vecext.2) |
| 61 | + %vecins.2 = insertelement <4 x i64> %vecins.1, i64 %3, i64 2 |
| 62 | + %vecext.3 = extractelement <4 x float> %0, i64 3 |
| 63 | + %4 = call i64 @llvm.lrint.i64.f32(float %vecext.3) |
| 64 | + %vecins.3 = insertelement <4 x i64> %vecins.2, i64 %4, i64 3 |
| 65 | + ret <4 x i64> %vecins.3 |
| 66 | +} |
| 67 | + |
| 68 | +declare float @llvm.rint.f32(float) |
| 69 | +declare i64 @llvm.lrint.i64.f32(float) |
0 commit comments