87 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 52; RUN: opt -p loop-vectorize -S %s | FileCheck %s3 4target triple = "arm64-apple-macosx"5 6; Test case for https://github.com/llvm/llvm-project/issues/107015.7define i64 @mul_select_operand_known_1_via_scev() {8; CHECK-LABEL: define i64 @mul_select_operand_known_1_via_scev() {9; CHECK-NEXT: [[ENTRY:.*]]:10; CHECK-NEXT: br label %[[LOOP:.*]]11; CHECK: [[LOOP]]:12; CHECK-NEXT: [[RED:%.*]] = phi i64 [ 12, %[[ENTRY]] ], [ [[RED_NEXT:%.*]], %[[LOOP]] ]13; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP]] ]14; CHECK-NEXT: [[TMP1:%.*]] = and i32 [[IV]], 115; CHECK-NEXT: [[CMP1_I:%.*]] = icmp eq i32 [[TMP1]], 016; CHECK-NEXT: [[NARROW_I:%.*]] = select i1 [[CMP1_I]], i32 1, i32 [[IV]]17; CHECK-NEXT: [[MUL:%.*]] = zext nneg i32 [[NARROW_I]] to i6418; CHECK-NEXT: [[RED_NEXT]] = mul nsw i64 [[RED]], [[MUL]]19; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i32 [[IV]], 120; CHECK-NEXT: [[EC:%.*]] = icmp eq i32 [[IV]], 121; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]22; CHECK: [[EXIT]]:23; CHECK-NEXT: [[RES:%.*]] = phi i64 [ [[RED_NEXT]], %[[LOOP]] ]24; CHECK-NEXT: ret i64 [[RES]]25;26entry:27 br label %loop28 29loop:30 %red = phi i64 [ 12, %entry ], [ %red.next, %loop ]31 %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop ]32 %0 = and i32 %iv, 133 %cmp1.i = icmp eq i32 %0, 034 %narrow.i = select i1 %cmp1.i, i32 1, i32 %iv35 %mul = zext nneg i32 %narrow.i to i6436 %red.next = mul nsw i64 %red, %mul37 %iv.next = add nuw nsw i32 %iv, 138 %ec = icmp eq i32 %iv, 139 br i1 %ec, label %exit, label %loop40 41exit:42 %res = phi i64 [ %red.next, %loop ]43 ret i64 %res44}45 46define i32 @add_reduction_select_operand_constant_but_non_uniform() {47; CHECK-LABEL: define i32 @add_reduction_select_operand_constant_but_non_uniform() {48; CHECK-NEXT: [[ENTRY:.*:]]49; CHECK-NEXT: br label %[[VECTOR_PH:.*]]50; CHECK: [[VECTOR_PH]]:51; CHECK-NEXT: br label %[[VECTOR_BODY:.*]]52; CHECK: [[VECTOR_BODY]]:53; CHECK-NEXT: [[INDEX:%.*]] = phi i32 [ 0, %[[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], %[[VECTOR_BODY]] ]54; CHECK-NEXT: [[TMP2:%.*]] = phi <4 x i32> [ <i32 42, i32 0, i32 0, i32 0>, %[[VECTOR_PH]] ], [ [[TMP2]], %[[VECTOR_BODY]] ]55; CHECK-NEXT: [[TMP1:%.*]] = phi <4 x i32> [ zeroinitializer, %[[VECTOR_PH]] ], [ [[TMP1]], %[[VECTOR_BODY]] ]56; CHECK-NEXT: [[INDEX_NEXT]] = add nuw i32 [[INDEX]], 857; CHECK-NEXT: [[TMP0:%.*]] = icmp eq i32 [[INDEX_NEXT]], 6458; CHECK-NEXT: br i1 [[TMP0]], label %[[MIDDLE_BLOCK:.*]], label %[[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]59; CHECK: [[MIDDLE_BLOCK]]:60; CHECK-NEXT: [[BIN_RDX:%.*]] = add <4 x i32> [[TMP1]], [[TMP2]]61; CHECK-NEXT: [[TMP3:%.*]] = call i32 @llvm.vector.reduce.add.v4i32(<4 x i32> [[BIN_RDX]])62; CHECK-NEXT: br label %[[EXIT:.*]]63; CHECK: [[EXIT]]:64; CHECK-NEXT: ret i32 [[TMP3]]65;66entry:67 br label %loop68 69loop:70 %iv = phi i32 [ 0, %entry ], [ %iv.next, %loop ]71 %rdx = phi i32 [ 42, %entry ], [ %rdx.next, %loop ]72 73 %iv.next = add i32 %iv, 174 %rdx.next = add i32 0, %rdx75 76 %cmp = icmp ult i32 %iv.next, 6477 br i1 %cmp, label %loop, label %exit78 79exit:80 ret i32 %rdx.next81}82;.83; CHECK: [[LOOP0]] = distinct !{[[LOOP0]], [[META1:![0-9]+]], [[META2:![0-9]+]]}84; CHECK: [[META1]] = !{!"llvm.loop.isvectorized", i32 1}85; CHECK: [[META2]] = !{!"llvm.loop.unroll.runtime.disable"}86;.87