1f269ec23SSjoerd Meijer; NOTE: Assertions have been autogenerated by utils/update_test_checks.py 2*d1e5e673SSimon Pilgrim; RUN: opt -passes="default<O3>" -enable-loop-flatten -loop-flatten-cost-threshold=3 -S %s | FileCheck %s 3f269ec23SSjoerd Meijer 4f269ec23SSjoerd Meijertarget datalayout = "e-m:e-i8:8:32-i16:16:32-i64:64-i128:128-n32:64-S128" 5f269ec23SSjoerd Meijertarget triple = "aarch64" 6f269ec23SSjoerd Meijer 71c55cc60SMatt Arsenaultdefine dso_local void @_Z3fooPiii(ptr %A, i32 %N, i32 %M) #0 { 8f269ec23SSjoerd Meijer; CHECK-LABEL: @_Z3fooPiii( 9f269ec23SSjoerd Meijer; CHECK-NEXT: entry: 103b823361Sdewen; CHECK-NEXT: [[CMP3:%.*]] = icmp sgt i32 [[N:%.*]], 0 113b823361Sdewen; CHECK-NEXT: [[CMP21:%.*]] = icmp sgt i32 [[M:%.*]], 0 123b823361Sdewen; CHECK-NEXT: [[OR_COND:%.*]] = select i1 [[CMP3]], i1 [[CMP21]], i1 false 133b823361Sdewen; CHECK-NEXT: br i1 [[OR_COND]], label [[FOR_COND1_PREHEADER_LR_PH_SPLIT_US:%.*]], label [[FOR_COND_CLEANUP:%.*]] 143b823361Sdewen; CHECK: for.cond1.preheader.lr.ph.split.us: 153b823361Sdewen; CHECK-NEXT: [[TMP0:%.*]] = zext nneg i32 [[M]] to i64 163b823361Sdewen; CHECK-NEXT: [[TMP1:%.*]] = zext nneg i32 [[N]] to i64 173b823361Sdewen; CHECK-NEXT: [[FLATTEN_TRIPCOUNT:%.*]] = mul nuw nsw i64 [[TMP0]], [[TMP1]] 183b823361Sdewen; CHECK-NEXT: br label [[FOR_COND1_PREHEADER_US:%.*]] 19f269ec23SSjoerd Meijer; CHECK: for.cond1.preheader.us: 203b823361Sdewen; CHECK-NEXT: [[INDVAR6:%.*]] = phi i64 [ [[INDVAR_NEXT7:%.*]], [[FOR_COND1_PREHEADER_US]] ], [ 0, [[FOR_COND1_PREHEADER_LR_PH_SPLIT_US]] ] 21462cb3cdSNikita Popov; CHECK-NEXT: [[ARRAYIDX_US:%.*]] = getelementptr inbounds nuw i32, ptr [[A:%.*]], i64 [[INDVAR6]] 223b823361Sdewen; CHECK-NEXT: [[TMP2:%.*]] = load i32, ptr [[ARRAYIDX_US]], align 4 23f269ec23SSjoerd Meijer; CHECK-NEXT: tail call void @_Z1fi(i32 [[TMP2]]) 24dce77a35SFlorian Hahn; CHECK-NEXT: [[INDVAR_NEXT7]] = add nuw i64 [[INDVAR6]], 1 253b823361Sdewen; CHECK-NEXT: [[EXITCOND_NOT:%.*]] = icmp eq i64 [[INDVAR_NEXT7]], [[FLATTEN_TRIPCOUNT]] 263b823361Sdewen; CHECK-NEXT: br i1 [[EXITCOND_NOT]], label [[FOR_COND_CLEANUP]], label [[FOR_COND1_PREHEADER_US]] 27f269ec23SSjoerd Meijer; CHECK: for.cond.cleanup: 28f269ec23SSjoerd Meijer; CHECK-NEXT: ret void 29f269ec23SSjoerd Meijer; 30f269ec23SSjoerd Meijerentry: 31f269ec23SSjoerd Meijer br label %for.cond 32f269ec23SSjoerd Meijer 33f269ec23SSjoerd Meijerfor.cond: 34f269ec23SSjoerd Meijer %i.0 = phi i32 [ 0, %entry ], [ %inc6, %for.cond.cleanup3 ] 35f269ec23SSjoerd Meijer %cmp = icmp slt i32 %i.0, %N 36f269ec23SSjoerd Meijer br i1 %cmp, label %for.body, label %for.cond.cleanup 37f269ec23SSjoerd Meijer 38f269ec23SSjoerd Meijerfor.cond.cleanup: 39f269ec23SSjoerd Meijer ret void 40f269ec23SSjoerd Meijer 41f269ec23SSjoerd Meijerfor.body: 42f269ec23SSjoerd Meijer br label %for.cond1 43f269ec23SSjoerd Meijer 44f269ec23SSjoerd Meijerfor.cond1: 45f269ec23SSjoerd Meijer %j.0 = phi i32 [ 0, %for.body ], [ %inc, %for.body4 ] 46f269ec23SSjoerd Meijer %cmp2 = icmp slt i32 %j.0, %M 47f269ec23SSjoerd Meijer br i1 %cmp2, label %for.body4, label %for.cond.cleanup3 48f269ec23SSjoerd Meijer 49f269ec23SSjoerd Meijerfor.cond.cleanup3: 50f269ec23SSjoerd Meijer %inc6 = add nsw i32 %i.0, 1 51f269ec23SSjoerd Meijer br label %for.cond 52f269ec23SSjoerd Meijer 53f269ec23SSjoerd Meijerfor.body4: 54f269ec23SSjoerd Meijer %mul = mul nsw i32 %i.0, %M 55f269ec23SSjoerd Meijer %add = add nsw i32 %mul, %j.0 56f269ec23SSjoerd Meijer %idxprom = sext i32 %add to i64 571c55cc60SMatt Arsenault %arrayidx = getelementptr inbounds i32, ptr %A, i64 %idxprom 581c55cc60SMatt Arsenault %0 = load i32, ptr %arrayidx, align 4 59f269ec23SSjoerd Meijer call void @_Z1fi(i32 %0) 60f269ec23SSjoerd Meijer %inc = add nsw i32 %j.0, 1 61f269ec23SSjoerd Meijer br label %for.cond1 62f269ec23SSjoerd Meijer} 63f269ec23SSjoerd Meijer 64f269ec23SSjoerd Meijerdeclare dso_local void @_Z1fi(i32) #2 65