xref: /llvm-project/llvm/test/Transforms/LoopVectorize/is_fpclass.ll (revision 1de3dc7d23dd6b856efad3a3a04f2396328726d7)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2; RUN: opt < %s -passes=loop-vectorize -force-vector-width=2 -S | FileCheck %s
3
4define void @d() {
5; CHECK-LABEL: define void @d() {
6; CHECK-NEXT:  entry:
7; CHECK-NEXT:    br i1 false, label [[SCALAR_PH:%.*]], label [[VECTOR_PH:%.*]]
8; CHECK:       vector.ph:
9; CHECK-NEXT:    br label [[VECTOR_BODY:%.*]]
10; CHECK:       vector.body:
11; CHECK-NEXT:    [[INDEX:%.*]] = phi i64 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ]
12; CHECK-NEXT:    [[TMP0:%.*]] = add i64 [[INDEX]], 0
13; CHECK-NEXT:    [[TMP6:%.*]] = load float, ptr null, align 4
14; CHECK-NEXT:    [[BROADCAST_SPLATINSERT:%.*]] = insertelement <2 x float> poison, float [[TMP6]], i64 0
15; CHECK-NEXT:    [[BROADCAST_SPLAT:%.*]] = shufflevector <2 x float> [[BROADCAST_SPLATINSERT]], <2 x float> poison, <2 x i32> zeroinitializer
16; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr float, ptr @d, i64 [[TMP0]]
17; CHECK-NEXT:    [[TMP2:%.*]] = call <2 x i1> @llvm.is.fpclass.v2f32(<2 x float> [[BROADCAST_SPLAT]], i32 0)
18; CHECK-NEXT:    [[TMP3:%.*]] = select <2 x i1> [[TMP2]], <2 x float> zeroinitializer, <2 x float> zeroinitializer
19; CHECK-NEXT:    [[TMP4:%.*]] = getelementptr float, ptr [[TMP1]], i32 0
20; CHECK-NEXT:    store <2 x float> [[TMP3]], ptr [[TMP4]], align 4
21; CHECK-NEXT:    [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 2
22; CHECK-NEXT:    [[TMP7:%.*]] = icmp eq i64 [[INDEX_NEXT]], 128
23; CHECK-NEXT:    br i1 [[TMP7]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]
24; CHECK:       middle.block:
25; CHECK-NEXT:    br i1 true, label [[EXIT:%.*]], label [[SCALAR_PH]]
26; CHECK:       scalar.ph:
27; CHECK-NEXT:    [[BC_RESUME_VAL:%.*]] = phi i64 [ 128, [[MIDDLE_BLOCK]] ], [ 0, [[ENTRY:%.*]] ]
28; CHECK-NEXT:    br label [[LOOP:%.*]]
29; CHECK:       loop:
30; CHECK-NEXT:    [[I:%.*]] = phi i64 [ [[BC_RESUME_VAL]], [[SCALAR_PH]] ], [ [[I7:%.*]], [[LOOP]] ]
31; CHECK-NEXT:    [[I3:%.*]] = load float, ptr null, align 4
32; CHECK-NEXT:    [[I4:%.*]] = getelementptr float, ptr @d, i64 [[I]]
33; CHECK-NEXT:    [[I5:%.*]] = tail call i1 @llvm.is.fpclass.f32(float [[I3]], i32 0)
34; CHECK-NEXT:    [[I6:%.*]] = select i1 [[I5]], float 0.000000e+00, float 0.000000e+00
35; CHECK-NEXT:    store float [[I6]], ptr [[I4]], align 4
36; CHECK-NEXT:    [[I7]] = add i64 [[I]], 1
37; CHECK-NEXT:    [[I8:%.*]] = icmp eq i64 [[I7]], 128
38; CHECK-NEXT:    br i1 [[I8]], label [[EXIT]], label [[LOOP]], !llvm.loop [[LOOP3:![0-9]+]]
39; CHECK:       exit:
40; CHECK-NEXT:    ret void
41;
42entry:
43  br label %loop
44
45loop:
46  %i = phi i64 [ 0, %entry ], [ %i7, %loop ]
47  %i3 = load float, ptr null, align 4
48  %i4 = getelementptr float, ptr @d, i64 %i
49  %i5 = tail call i1 @llvm.is.fpclass.f32(float %i3, i32 0)
50  %i6 = select i1 %i5, float 0.0, float 0.0
51  store float %i6, ptr %i4, align 4
52  %i7 = add i64 %i, 1
53  %i8 = icmp eq i64 %i7, 128
54  br i1 %i8, label %exit, label %loop
55
56exit:
57  ret void
58}
59
60declare i1 @llvm.is.fpclass.f32(float, i32 immarg)
61