|
| 1 | +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5 |
| 2 | +; RUN: opt -passes=loop-vectorize -force-vector-width=2 -mtriple=riscv64 -mattr=+v -S %s | FileCheck %s |
| 3 | +define void @truncate_i16_to_i8_cse(ptr noalias %src, ptr noalias %dst) { |
| 4 | +; CHECK-LABEL: define void @truncate_i16_to_i8_cse( |
| 5 | +; CHECK-SAME: ptr noalias [[SRC:%.*]], ptr noalias [[DST:%.*]]) #[[ATTR0:[0-9]+]] { |
| 6 | +; CHECK-NEXT: [[ENTRY:.*:]] |
| 7 | +; CHECK-NEXT: br label %[[VECTOR_PH:.*]] |
| 8 | +; CHECK: [[VECTOR_PH]]: |
| 9 | +; CHECK-NEXT: [[BROADCAST_SPLATINSERT1:%.*]] = insertelement <2 x ptr> poison, ptr [[DST]], i64 0 |
| 10 | +; CHECK-NEXT: [[BROADCAST_SPLAT1:%.*]] = shufflevector <2 x ptr> [[BROADCAST_SPLATINSERT1]], <2 x ptr> poison, <2 x i32> zeroinitializer |
| 11 | +; CHECK-NEXT: br label %[[VECTOR_BODY:.*]] |
| 12 | +; CHECK: [[VECTOR_BODY]]: |
| 13 | +; CHECK-NEXT: [[INDEX:%.*]] = phi i64 [ 0, %[[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], %[[VECTOR_BODY]] ] |
| 14 | +; CHECK-NEXT: [[TMP0:%.*]] = load i16, ptr [[SRC]], align 2 |
| 15 | +; CHECK-NEXT: [[BROADCAST_SPLATINSERT:%.*]] = insertelement <2 x i16> poison, i16 [[TMP0]], i64 0 |
| 16 | +; CHECK-NEXT: [[BROADCAST_SPLAT:%.*]] = shufflevector <2 x i16> [[BROADCAST_SPLATINSERT]], <2 x i16> poison, <2 x i32> zeroinitializer |
| 17 | +; CHECK-NEXT: [[TMP1:%.*]] = trunc <2 x i16> [[BROADCAST_SPLAT]] to <2 x i8> |
| 18 | +; CHECK-NEXT: call void @llvm.masked.scatter.v2i8.v2p0(<2 x i8> [[TMP1]], <2 x ptr> align 1 zeroinitializer, <2 x i1> splat (i1 true)) |
| 19 | +; CHECK-NEXT: call void @llvm.masked.scatter.v2i8.v2p0(<2 x i8> [[TMP1]], <2 x ptr> align 1 [[BROADCAST_SPLAT1]], <2 x i1> splat (i1 true)) |
| 20 | +; CHECK-NEXT: [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 2 |
| 21 | +; CHECK-NEXT: [[TMP3:%.*]] = icmp eq i64 [[INDEX_NEXT]], 4294967296 |
| 22 | +; CHECK-NEXT: br i1 [[TMP3]], label %[[MIDDLE_BLOCK:.*]], label %[[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]] |
| 23 | +; CHECK: [[MIDDLE_BLOCK]]: |
| 24 | +; CHECK-NEXT: br label %[[EXIT:.*]] |
| 25 | +; CHECK: [[EXIT]]: |
| 26 | +; CHECK-NEXT: ret void |
| 27 | +; |
| 28 | +entry: |
| 29 | + br label %loop |
| 30 | + |
| 31 | +loop: |
| 32 | + %iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ] |
| 33 | + %count = phi i32 [ 0, %entry ], [ %count.next, %loop ] |
| 34 | + %val = load i16, ptr %src, align 2 |
| 35 | + %val.zext = zext i16 %val to i64 |
| 36 | + %val.trunc.zext = trunc i64 %val.zext to i8 |
| 37 | + store i8 %val.trunc.zext, ptr null, align 1 |
| 38 | + %val.trunc = trunc i16 %val to i8 |
| 39 | + store i8 %val.trunc, ptr %dst, align 1 |
| 40 | + %count.next = add i32 %count, 1 |
| 41 | + %exitcond = icmp eq i32 %count.next, 0 |
| 42 | + %iv.next = add i64 %iv, 1 |
| 43 | + br i1 %exitcond, label %exit, label %loop |
| 44 | + |
| 45 | +exit: |
| 46 | + ret void |
| 47 | +} |
| 48 | +;. |
| 49 | +; CHECK: [[LOOP0]] = distinct !{[[LOOP0]], [[META1:![0-9]+]], [[META2:![0-9]+]]} |
| 50 | +; CHECK: [[META1]] = !{!"llvm.loop.isvectorized", i32 1} |
| 51 | +; CHECK: [[META2]] = !{!"llvm.loop.unroll.runtime.disable"} |
| 52 | +;. |
0 commit comments