|
| 1 | +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4 |
| 2 | +; RUN: opt -passes=loop-vectorize -force-vector-interleave=1 -S < %s | FileCheck %s |
| 3 | + |
| 4 | +target datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128" |
| 5 | +target triple = "aarch64-none-unknown-elf" |
| 6 | + |
| 7 | +define void @dotp(ptr %a, ptr %b) #0 { |
| 8 | +; CHECK-LABEL: define void @dotp( |
| 9 | +; CHECK-SAME: ptr [[A:%.*]], ptr [[B:%.*]]) #[[ATTR0:[0-9]+]] { |
| 10 | +; CHECK-NEXT: entry: |
| 11 | +; CHECK-NEXT: [[TMP0:%.*]] = call i64 @llvm.vscale.i64() |
| 12 | +; CHECK-NEXT: [[TMP1:%.*]] = mul i64 [[TMP0]], 16 |
| 13 | +; CHECK-NEXT: [[MIN_ITERS_CHECK:%.*]] = icmp ult i64 0, [[TMP1]] |
| 14 | +; CHECK-NEXT: br i1 [[MIN_ITERS_CHECK]], label [[SCALAR_PH:%.*]], label [[VECTOR_PH:%.*]] |
| 15 | +; CHECK: vector.ph: |
| 16 | +; CHECK-NEXT: [[TMP2:%.*]] = call i64 @llvm.vscale.i64() |
| 17 | +; CHECK-NEXT: [[TMP3:%.*]] = mul i64 [[TMP2]], 16 |
| 18 | +; CHECK-NEXT: [[N_MOD_VF:%.*]] = urem i64 0, [[TMP3]] |
| 19 | +; CHECK-NEXT: [[N_VEC:%.*]] = sub i64 0, [[N_MOD_VF]] |
| 20 | +; CHECK-NEXT: [[TMP4:%.*]] = call i64 @llvm.vscale.i64() |
| 21 | +; CHECK-NEXT: [[TMP5:%.*]] = mul i64 [[TMP4]], 16 |
| 22 | +; CHECK-NEXT: br label [[VECTOR_BODY:%.*]] |
| 23 | +; CHECK: vector.body: |
| 24 | +; CHECK-NEXT: [[INDEX:%.*]] = phi i64 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ] |
| 25 | +; CHECK-NEXT: [[VEC_PHI:%.*]] = phi <vscale x 16 x i32> [ zeroinitializer, [[VECTOR_PH]] ], [ [[TMP14:%.*]], [[VECTOR_BODY]] ] |
| 26 | +; CHECK-NEXT: [[TMP11:%.*]] = add i64 [[INDEX]], 0 |
| 27 | +; CHECK-NEXT: [[TMP13:%.*]] = getelementptr i8, ptr [[A]], i64 [[TMP11]] |
| 28 | +; CHECK-NEXT: [[TMP17:%.*]] = getelementptr i8, ptr [[TMP13]], i32 0 |
| 29 | +; CHECK-NEXT: [[WIDE_LOAD2:%.*]] = load <vscale x 16 x i8>, ptr [[TMP17]], align 1 |
| 30 | +; CHECK-NEXT: [[TMP19:%.*]] = zext <vscale x 16 x i8> [[WIDE_LOAD2]] to <vscale x 16 x i32> |
| 31 | +; CHECK-NEXT: [[TMP21:%.*]] = getelementptr i8, ptr [[B]], i64 [[TMP11]] |
| 32 | +; CHECK-NEXT: [[TMP25:%.*]] = getelementptr i8, ptr [[TMP21]], i32 0 |
| 33 | +; CHECK-NEXT: [[WIDE_LOAD4:%.*]] = load <vscale x 16 x i8>, ptr [[TMP25]], align 1 |
| 34 | +; CHECK-NEXT: [[TMP27:%.*]] = zext <vscale x 16 x i8> [[WIDE_LOAD4]] to <vscale x 16 x i32> |
| 35 | +; CHECK-NEXT: [[TMP29:%.*]] = mul <vscale x 16 x i32> [[TMP27]], [[TMP19]] |
| 36 | +; CHECK-NEXT: [[TMP14]] = add <vscale x 16 x i32> [[TMP29]], [[VEC_PHI]] |
| 37 | +; CHECK-NEXT: [[INDEX_NEXT]] = add nuw i64 [[INDEX]], [[TMP5]] |
| 38 | +; CHECK-NEXT: [[TMP32:%.*]] = icmp eq i64 [[INDEX_NEXT]], [[N_VEC]] |
| 39 | +; CHECK-NEXT: br i1 [[TMP32]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]] |
| 40 | +; CHECK: middle.block: |
| 41 | +; CHECK-NEXT: [[TMP33:%.*]] = call i32 @llvm.vector.reduce.add.nxv16i32(<vscale x 16 x i32> [[TMP14]]) |
| 42 | +; CHECK-NEXT: [[CMP_N:%.*]] = icmp eq i64 0, [[N_VEC]] |
| 43 | +; CHECK-NEXT: br i1 [[CMP_N]], label [[FOR_COND_CLEANUP_LOOPEXIT:%.*]], label [[SCALAR_PH]] |
| 44 | +; CHECK: scalar.ph: |
| 45 | +; CHECK-NEXT: [[BC_RESUME_VAL:%.*]] = phi i64 [ [[N_VEC]], [[MIDDLE_BLOCK]] ], [ 0, [[ENTRY:%.*]] ] |
| 46 | +; CHECK-NEXT: [[BC_MERGE_RDX:%.*]] = phi i32 [ 0, [[ENTRY]] ], [ [[TMP33]], [[MIDDLE_BLOCK]] ] |
| 47 | +; CHECK-NEXT: br label [[FOR_BODY:%.*]] |
| 48 | +; CHECK: for.cond.cleanup.loopexit: |
| 49 | +; CHECK-NEXT: [[ADD_LCSSA:%.*]] = phi i32 [ [[ADD:%.*]], [[FOR_BODY]] ], [ [[TMP33]], [[MIDDLE_BLOCK]] ] |
| 50 | +; CHECK-NEXT: [[TMP20:%.*]] = lshr i32 [[ADD_LCSSA]], 0 |
| 51 | +; CHECK-NEXT: ret void |
| 52 | +; CHECK: for.body: |
| 53 | +; CHECK-NEXT: [[INDVARS_IV:%.*]] = phi i64 [ [[BC_RESUME_VAL]], [[SCALAR_PH]] ], [ [[INDVARS_IV_NEXT:%.*]], [[FOR_BODY]] ] |
| 54 | +; CHECK-NEXT: [[ACC_010:%.*]] = phi i32 [ [[BC_MERGE_RDX]], [[SCALAR_PH]] ], [ [[ADD]], [[FOR_BODY]] ] |
| 55 | +; CHECK-NEXT: [[ARRAYIDX:%.*]] = getelementptr i8, ptr [[A]], i64 [[INDVARS_IV]] |
| 56 | +; CHECK-NEXT: [[TMP18:%.*]] = load i8, ptr [[ARRAYIDX]], align 1 |
| 57 | +; CHECK-NEXT: [[CONV:%.*]] = zext i8 [[TMP18]] to i32 |
| 58 | +; CHECK-NEXT: [[ARRAYIDX2:%.*]] = getelementptr i8, ptr [[B]], i64 [[INDVARS_IV]] |
| 59 | +; CHECK-NEXT: [[TMP22:%.*]] = load i8, ptr [[ARRAYIDX2]], align 1 |
| 60 | +; CHECK-NEXT: [[CONV3:%.*]] = zext i8 [[TMP22]] to i32 |
| 61 | +; CHECK-NEXT: [[MUL:%.*]] = mul i32 [[CONV3]], [[CONV]] |
| 62 | +; CHECK-NEXT: [[ADD]] = add i32 [[MUL]], [[ACC_010]] |
| 63 | +; CHECK-NEXT: [[INDVARS_IV_NEXT]] = add i64 [[INDVARS_IV]], 1 |
| 64 | +; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i64 [[INDVARS_IV_NEXT]], 0 |
| 65 | +; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label [[FOR_COND_CLEANUP_LOOPEXIT]], label [[FOR_BODY]], !llvm.loop [[LOOP3:![0-9]+]] |
| 66 | +; |
| 67 | +entry: |
| 68 | + br label %for.body |
| 69 | + |
| 70 | +for.cond.cleanup.loopexit: ; preds = %for.body |
| 71 | + %0 = lshr i32 %add, 0 |
| 72 | + ret void |
| 73 | + |
| 74 | +for.body: ; preds = %for.body, %entry |
| 75 | + %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ] |
| 76 | + %acc.010 = phi i32 [ 0, %entry ], [ %add, %for.body ] |
| 77 | + %arrayidx = getelementptr i8, ptr %a, i64 %indvars.iv |
| 78 | + %1 = load i8, ptr %arrayidx, align 1 |
| 79 | + %conv = zext i8 %1 to i32 |
| 80 | + %arrayidx2 = getelementptr i8, ptr %b, i64 %indvars.iv |
| 81 | + %2 = load i8, ptr %arrayidx2, align 1 |
| 82 | + %conv3 = zext i8 %2 to i32 |
| 83 | + %mul = mul i32 %conv3, %conv |
| 84 | + %add = add i32 %mul, %acc.010 |
| 85 | + %indvars.iv.next = add i64 %indvars.iv, 1 |
| 86 | + %exitcond.not = icmp eq i64 %indvars.iv.next, 0 |
| 87 | + br i1 %exitcond.not, label %for.cond.cleanup.loopexit, label %for.body |
| 88 | + |
| 89 | +; uselistorder directives |
| 90 | + uselistorder i32 %add, { 1, 0 } |
| 91 | +} |
| 92 | + |
| 93 | +attributes #0 = { "target-features"="+fp-armv8,+fullfp16,+neon,+sve,+sve2,+v8a" } |
| 94 | +;. |
| 95 | +; CHECK: [[LOOP0]] = distinct !{[[LOOP0]], [[META1:![0-9]+]], [[META2:![0-9]+]]} |
| 96 | +; CHECK: [[META1]] = !{!"llvm.loop.isvectorized", i32 1} |
| 97 | +; CHECK: [[META2]] = !{!"llvm.loop.unroll.runtime.disable"} |
| 98 | +; CHECK: [[LOOP3]] = distinct !{[[LOOP3]], [[META2]], [[META1]]} |
| 99 | +;. |
0 commit comments