|
| 1 | +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals none --version 5 |
1 | 2 | ; RUN: opt -passes=loop-vectorize -force-vector-interleave=2 -force-vector-width=1 -S %s | FileCheck %s
|
2 | 3 |
|
3 | 4 | target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
|
4 | 5 |
|
5 |
| - |
6 | 6 | ; Test for PR54427.
|
7 | 7 | define void @test_nonconst_start_and_step(ptr %dst, i32 %start, i32 %step, i64 %N) {
|
8 |
| -; CHECK-LABEL: @test_nonconst_start_and_step( |
9 |
| -; CHECK: [[NEG_STEP:%.+]] = sub i32 0, %step |
10 |
| -; CHECK: vector.body: |
11 |
| -; CHECK-NEXT: [[INDEX:%.*]] = phi i64 [ 0, %vector.ph ], [ [[INDEX_NEXT:%.*]], %vector.body ] |
| 8 | +; CHECK-LABEL: define void @test_nonconst_start_and_step( |
| 9 | +; CHECK-SAME: ptr [[DST:%.*]], i32 [[START:%.*]], i32 [[STEP:%.*]], i64 [[N:%.*]]) { |
| 10 | +; CHECK-NEXT: [[ENTRY:.*]]: |
| 11 | +; CHECK-NEXT: [[NEG_STEP:%.*]] = sub i32 0, [[STEP]] |
| 12 | +; CHECK-NEXT: [[MIN_ITERS_CHECK:%.*]] = icmp ult i64 [[N]], 2 |
| 13 | +; CHECK-NEXT: br i1 [[MIN_ITERS_CHECK]], label %[[SCALAR_PH:.*]], label %[[VECTOR_PH:.*]] |
| 14 | +; CHECK: [[VECTOR_PH]]: |
| 15 | +; CHECK-NEXT: [[N_MOD_VF:%.*]] = urem i64 [[N]], 2 |
| 16 | +; CHECK-NEXT: [[N_VEC:%.*]] = sub i64 [[N]], [[N_MOD_VF]] |
| 17 | +; CHECK-NEXT: [[DOTCAST:%.*]] = trunc i64 [[N_VEC]] to i32 |
| 18 | +; CHECK-NEXT: [[TMP1:%.*]] = mul i32 [[DOTCAST]], [[NEG_STEP]] |
| 19 | +; CHECK-NEXT: [[TMP4:%.*]] = add i32 [[START]], [[TMP1]] |
| 20 | +; CHECK-NEXT: br label %[[VECTOR_BODY:.*]] |
| 21 | +; CHECK: [[VECTOR_BODY]]: |
| 22 | +; CHECK-NEXT: [[INDEX:%.*]] = phi i64 [ 0, %[[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], %[[VECTOR_BODY]] ] |
12 | 23 | ; CHECK-NEXT: [[INDUCTION4:%.*]] = add i64 [[INDEX]], 1
|
13 | 24 | ; CHECK-NEXT: [[TMP2:%.*]] = trunc i64 [[INDEX]] to i32
|
14 | 25 | ; CHECK-NEXT: [[TMP3:%.*]] = mul i32 [[TMP2]], [[NEG_STEP]]
|
15 |
| -; CHECK-NEXT: [[OFFSET_IDX:%.*]] = add i32 %start, [[TMP3]] |
| 26 | +; CHECK-NEXT: [[OFFSET_IDX:%.*]] = add i32 [[START]], [[TMP3]] |
16 | 27 | ; CHECK-NEXT: [[TMP5:%.*]] = mul i32 1, [[NEG_STEP]]
|
17 | 28 | ; CHECK-NEXT: [[INDUCTION2:%.*]] = add i32 [[OFFSET_IDX]], [[TMP5]]
|
18 |
| -; CHECK-NEXT: [[TMP6:%.*]] = sub nsw i32 [[OFFSET_IDX]], %step |
19 |
| -; CHECK-NEXT: [[TMP7:%.*]] = sub nsw i32 [[INDUCTION2]], %step |
20 |
| -; CHECK-NEXT: [[TMP8:%.*]] = getelementptr inbounds i32, ptr [[DST:%.*]], i64 [[INDEX]] |
| 29 | +; CHECK-NEXT: [[TMP6:%.*]] = sub nsw i32 [[OFFSET_IDX]], [[STEP]] |
| 30 | +; CHECK-NEXT: [[TMP7:%.*]] = sub nsw i32 [[INDUCTION2]], [[STEP]] |
| 31 | +; CHECK-NEXT: [[TMP8:%.*]] = getelementptr inbounds i32, ptr [[DST]], i64 [[INDEX]] |
21 | 32 | ; CHECK-NEXT: [[TMP9:%.*]] = getelementptr inbounds i32, ptr [[DST]], i64 [[INDUCTION4]]
|
22 | 33 | ; CHECK-NEXT: store i32 [[TMP6]], ptr [[TMP8]], align 2
|
23 | 34 | ; CHECK-NEXT: store i32 [[TMP7]], ptr [[TMP9]], align 2
|
24 | 35 | ; CHECK-NEXT: [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 2
|
25 |
| -; CHECK-NEXT: [[TMP10:%.*]] = icmp eq i64 [[INDEX_NEXT]] |
26 |
| -; CHECK-NEXT: br i1 [[TMP10]], label %middle.block, label %vector.body |
| 36 | +; CHECK-NEXT: [[TMP11:%.*]] = icmp eq i64 [[INDEX_NEXT]], [[N_VEC]] |
| 37 | +; CHECK-NEXT: br i1 [[TMP11]], label %[[MIDDLE_BLOCK:.*]], label %[[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]] |
| 38 | +; CHECK: [[MIDDLE_BLOCK]]: |
| 39 | +; CHECK-NEXT: [[CMP_N:%.*]] = icmp eq i64 [[N]], [[N_VEC]] |
| 40 | +; CHECK-NEXT: br i1 [[CMP_N]], label %[[EXIT:.*]], label %[[SCALAR_PH]] |
| 41 | +; CHECK: [[SCALAR_PH]]: |
| 42 | +; CHECK-NEXT: [[BC_RESUME_VAL:%.*]] = phi i64 [ [[N_VEC]], %[[MIDDLE_BLOCK]] ], [ 0, %[[ENTRY]] ] |
| 43 | +; CHECK-NEXT: [[BC_RESUME_VAL2:%.*]] = phi i32 [ [[TMP4]], %[[MIDDLE_BLOCK]] ], [ [[START]], %[[ENTRY]] ] |
| 44 | +; CHECK-NEXT: br label %[[LOOP:.*]] |
| 45 | +; CHECK: [[LOOP]]: |
| 46 | +; CHECK-NEXT: [[PRIMARY_IV:%.*]] = phi i64 [ [[BC_RESUME_VAL]], %[[SCALAR_PH]] ], [ [[PRIMARY_IV_NEXT:%.*]], %[[LOOP]] ] |
| 47 | +; CHECK-NEXT: [[IV_DOWN:%.*]] = phi i32 [ [[BC_RESUME_VAL2]], %[[SCALAR_PH]] ], [ [[IV_DOWN_NEXT:%.*]], %[[LOOP]] ] |
| 48 | +; CHECK-NEXT: [[IV_DOWN_NEXT]] = sub nsw i32 [[IV_DOWN]], [[STEP]] |
| 49 | +; CHECK-NEXT: [[GEP_DST:%.*]] = getelementptr inbounds i32, ptr [[DST]], i64 [[PRIMARY_IV]] |
| 50 | +; CHECK-NEXT: store i32 [[IV_DOWN_NEXT]], ptr [[GEP_DST]], align 2 |
| 51 | +; CHECK-NEXT: [[PRIMARY_IV_NEXT]] = add nuw nsw i64 [[PRIMARY_IV]], 1 |
| 52 | +; CHECK-NEXT: [[EXITCOND:%.*]] = icmp eq i64 [[PRIMARY_IV_NEXT]], [[N]] |
| 53 | +; CHECK-NEXT: br i1 [[EXITCOND]], label %[[EXIT]], label %[[LOOP]], !llvm.loop [[LOOP3:![0-9]+]] |
| 54 | +; CHECK: [[EXIT]]: |
| 55 | +; CHECK-NEXT: ret void |
27 | 56 | ;
|
28 | 57 | entry:
|
29 | 58 | br label %loop
|
|
0 commit comments