Skip to content

Commit a06be8a

Browse files
authored
SLP/RISCV: add negative test for lrint (#55208) (#65611)
The issue #55208 describes a current deficiency of the SLPVectorizer, namely that it doesn't vectorize code written with lrint, while similar code written with rint is vectorized. Add a test corresponding to this issue for the RISC-V target.
1 parent f98ca79 commit a06be8a

File tree

1 file changed

+69
-0
lines changed
  • llvm/test/Transforms/SLPVectorizer/RISCV

1 file changed

+69
-0
lines changed
Lines changed: 69 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,69 @@
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

Comments
 (0)