From 3e60d216a4829a0e9a5045afa3e060f89ea18b04 Mon Sep 17 00:00:00 2001 From: Florian Hahn Date: Wed, 1 Sep 2021 15:32:59 +0200 Subject: [PATCH] [LoopDistribute] Add tests inspired by PR50296, PR50288. --- .../LoopDistribute/pointer-phi-in-loop.ll | 100 +++++++++++++++++++++ 1 file changed, 100 insertions(+) create mode 100644 llvm/test/Transforms/LoopDistribute/pointer-phi-in-loop.ll diff --git a/llvm/test/Transforms/LoopDistribute/pointer-phi-in-loop.ll b/llvm/test/Transforms/LoopDistribute/pointer-phi-in-loop.ll new file mode 100644 index 0000000..89c2848 --- /dev/null +++ b/llvm/test/Transforms/LoopDistribute/pointer-phi-in-loop.ll @@ -0,0 +1,100 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt -aa-pipeline=basic-aa -passes='loop-distribute' -enable-loop-distribute -S %s | FileCheck %s + +; Testcases inspired by PR50296, PR50288. + +define void @phi_load_store_distribute(i1 %c, i16* %A, i16* %B, i16* %C) { +; CHECK-LABEL: @phi_load_store_distribute( +; CHECK-NEXT: entry: +; CHECK-NEXT: br label [[FOR_BODY:%.*]] +; CHECK: for.body: +; CHECK-NEXT: [[IV:%.*]] = phi i16 [ 0, [[ENTRY:%.*]] ], [ [[IV_NEXT:%.*]], [[IF_END:%.*]] ] +; CHECK-NEXT: [[LV:%.*]] = load i16, i16* [[A:%.*]], align 1 +; CHECK-NEXT: store i16 [[LV]], i16* [[A]], align 1 +; CHECK-NEXT: br i1 [[C:%.*]], label [[IF_THEN:%.*]], label [[IF_END]] +; CHECK: if.then: +; CHECK-NEXT: [[LV2:%.*]] = load i16, i16* [[A]], align 1 +; CHECK-NEXT: br label [[IF_END]] +; CHECK: if.end: +; CHECK-NEXT: [[C_SINK:%.*]] = phi i16* [ [[B:%.*]], [[IF_THEN]] ], [ [[C:%.*]], [[FOR_BODY]] ] +; CHECK-NEXT: [[LV3:%.*]] = load i16, i16* [[C_SINK]], align 2 +; CHECK-NEXT: [[ADD:%.*]] = add i16 [[LV3]], 10 +; CHECK-NEXT: store i16 [[ADD]], i16* [[C_SINK]], align 1 +; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i16 [[IV]], 1 +; CHECK-NEXT: [[TOBOOL_NOT:%.*]] = icmp eq i16 [[IV_NEXT]], 1000 +; CHECK-NEXT: br i1 [[TOBOOL_NOT]], label [[FOR_END_LOOPEXIT:%.*]], label [[FOR_BODY]] +; CHECK: for.end.loopexit: +; CHECK-NEXT: ret void +; +entry: + br label %for.body + +for.body: ; preds = %if.end, %entry + %iv = phi i16 [ 0, %entry ], [ %iv.next, %if.end ] + %lv = load i16, i16* %A, align 1 + store i16 %lv, i16* %A, align 1 + br i1 %c, label %if.then, label %if.end + +if.then: ; preds = %for.body + %lv2 = load i16, i16* %A, align 1 + br label %if.end + +if.end: ; preds = %if.then, %for.body + %c.sink = phi i16* [ %B, %if.then ], [ %C, %for.body ] + %lv3 = load i16, i16* %c.sink + %add = add i16 %lv3, 10 + store i16 %add, i16* %c.sink, align 1 + %iv.next = add nuw nsw i16 %iv, 1 + %tobool.not = icmp eq i16 %iv.next, 1000 + br i1 %tobool.not, label %for.end.loopexit, label %for.body + +for.end.loopexit: ; preds = %if.end + ret void +} + +define void @phi_load_distribute(i1 %c, i16* %A, i16* %B, i16* %C) { +; CHECK-LABEL: @phi_load_distribute( +; CHECK-NEXT: entry: +; CHECK-NEXT: br label [[FOR_BODY:%.*]] +; CHECK: for.body: +; CHECK-NEXT: [[IV:%.*]] = phi i16 [ 0, [[ENTRY:%.*]] ], [ [[IV_NEXT:%.*]], [[IF_END:%.*]] ] +; CHECK-NEXT: [[LV:%.*]] = load i16, i16* [[A:%.*]], align 1 +; CHECK-NEXT: store i16 [[LV]], i16* [[A]], align 1 +; CHECK-NEXT: br i1 [[C:%.*]], label [[IF_THEN:%.*]], label [[IF_END]] +; CHECK: if.then: +; CHECK-NEXT: [[LV2:%.*]] = load i16, i16* [[A]], align 1 +; CHECK-NEXT: br label [[IF_END]] +; CHECK: if.end: +; CHECK-NEXT: [[C_SINK:%.*]] = phi i16* [ [[B:%.*]], [[IF_THEN]] ], [ [[C:%.*]], [[FOR_BODY]] ] +; CHECK-NEXT: [[LV3:%.*]] = load i16, i16* [[C_SINK]], align 2 +; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i16 [[IV]], 1 +; CHECK-NEXT: [[TOBOOL_NOT:%.*]] = icmp eq i16 [[IV_NEXT]], 1000 +; CHECK-NEXT: br i1 [[TOBOOL_NOT]], label [[FOR_END_LOOPEXIT:%.*]], label [[FOR_BODY]] +; CHECK: for.end.loopexit: +; CHECK-NEXT: ret void +; +entry: + br label %for.body + +for.body: ; preds = %if.end, %entry + %iv = phi i16 [ 0, %entry ], [ %iv.next, %if.end ] + %lv = load i16, i16* %A, align 1 + store i16 %lv, i16* %A, align 1 + br i1 %c, label %if.then, label %if.end + +if.then: ; preds = %for.body + %lv2 = load i16, i16* %A, align 1 + br label %if.end + +if.end: ; preds = %if.then, %for.body + %c.sink = phi i16* [ %B, %if.then ], [ %C, %for.body ] + %lv3 = load i16, i16* %c.sink + %iv.next = add nuw nsw i16 %iv, 1 + %tobool.not = icmp eq i16 %iv.next, 1000 + br i1 %tobool.not, label %for.end.loopexit, label %for.body + +for.end.loopexit: ; preds = %if.end + ret void +} + + -- 2.7.4