171 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py2; RUN: opt < %s -passes=nary-reassociate -S | FileCheck %s3 4declare i32 @llvm.smax.i32(i32 %a, i32 %b)5 6; m1 = smax(a,b) ; has side uses7; m2 = smax(smax((b,c), a) -> m2 = smax(m1, c)8define i32 @smax_test1(i32 %a, i32 %b, i32 %c) {9; CHECK-LABEL: @smax_test1(10; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[A:%.*]], [[B:%.*]]11; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]12; CHECK-NEXT: [[SMAX3_NARY:%.*]] = call i32 @llvm.smax.i32(i32 [[SMAX1]], i32 [[C:%.*]])13; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3_NARY]]14; CHECK-NEXT: ret i32 [[RES]]15;16 %c1 = icmp sgt i32 %a, %b17 %smax1 = select i1 %c1, i32 %a, i32 %b18 %c2 = icmp sgt i32 %b, %c19 %smax2 = select i1 %c2, i32 %b, i32 %c20 %c3 = icmp sgt i32 %smax2, %a21 %smax3 = select i1 %c3, i32 %smax2, i32 %a22 %res = add i32 %smax1, %smax323 ret i32 %res24}25 26; m1 = smax(a,b) ; has side uses27; m2 = smax(b, (smax(a, c))) -> m2 = smax(m1, c)28define i32 @smax_test2(i32 %a, i32 %b, i32 %c) {29; CHECK-LABEL: @smax_test2(30; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[A:%.*]], [[B:%.*]]31; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]32; CHECK-NEXT: [[SMAX3_NARY:%.*]] = call i32 @llvm.smax.i32(i32 [[SMAX1]], i32 [[C:%.*]])33; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3_NARY]]34; CHECK-NEXT: ret i32 [[RES]]35;36 %c1 = icmp sgt i32 %a, %b37 %smax1 = select i1 %c1, i32 %a, i32 %b38 %c2 = icmp sgt i32 %a, %c39 %smax2 = select i1 %c2, i32 %a, i32 %c40 %c3 = icmp sgt i32 %b, %smax241 %smax3 = select i1 %c3, i32 %b, i32 %smax242 %res = add i32 %smax1, %smax343 ret i32 %res44}45 46; Same test as smax_test1 but uses @llvm.smax intrinsic47define i32 @smax_test3(i32 %a, i32 %b, i32 %c) {48; CHECK-LABEL: @smax_test3(49; CHECK-NEXT: [[SMAX1:%.*]] = call i32 @llvm.smax.i32(i32 [[A:%.*]], i32 [[B:%.*]])50; CHECK-NEXT: [[SMAX3_NARY:%.*]] = call i32 @llvm.smax.i32(i32 [[SMAX1]], i32 [[C:%.*]])51; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3_NARY]]52; CHECK-NEXT: ret i32 [[RES]]53;54 %smax1 = call i32 @llvm.smax.i32(i32 %a, i32 %b)55 %smax2 = call i32 @llvm.smax.i32(i32 %b, i32 %c)56 %smax3 = call i32 @llvm.smax.i32(i32 %smax2, i32 %a)57 %res = add i32 %smax1, %smax358 ret i32 %res59}60 61; m1 = smax(a,b) ; has side uses62; m2 = smax(smax_or_eq((b,c), a) -> m2 = smax(m1, c)63define i32 @umax_test4(i32 %a, i32 %b, i32 %c) {64; CHECK-LABEL: @umax_test4(65; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[A:%.*]], [[B:%.*]]66; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]67; CHECK-NEXT: [[SMAX3_NARY:%.*]] = call i32 @llvm.smax.i32(i32 [[SMAX1]], i32 [[C:%.*]])68; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3_NARY]]69; CHECK-NEXT: ret i32 [[RES]]70;71 %c1 = icmp sgt i32 %a, %b72 %smax1 = select i1 %c1, i32 %a, i32 %b73 %c2 = icmp sge i32 %b, %c74 %smax_or_eq2 = select i1 %c2, i32 %b, i32 %c75 %c3 = icmp sgt i32 %smax_or_eq2, %a76 %smax3 = select i1 %c3, i32 %smax_or_eq2, i32 %a77 %res = add i32 %smax1, %smax378 ret i32 %res79}80 81; m1 = smax_or_eq(a,b) ; has side uses82; m2 = smax_or_eq(smax((b,c), a) -> m2 = smax(m1, c)83define i32 @smax_test5(i32 %a, i32 %b, i32 %c) {84; CHECK-LABEL: @smax_test5(85; CHECK-NEXT: [[C1:%.*]] = icmp sge i32 [[A:%.*]], [[B:%.*]]86; CHECK-NEXT: [[SMAX_OR_EQ1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]87; CHECK-NEXT: [[SMAX_OR_EQ3_NARY:%.*]] = call i32 @llvm.smax.i32(i32 [[SMAX_OR_EQ1]], i32 [[C:%.*]])88; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX_OR_EQ1]], [[SMAX_OR_EQ3_NARY]]89; CHECK-NEXT: ret i32 [[RES]]90;91 %c1 = icmp sge i32 %a, %b92 %smax_or_eq1 = select i1 %c1, i32 %a, i32 %b93 %c2 = icmp sgt i32 %b, %c94 %smax2 = select i1 %c2, i32 %b, i32 %c95 %c3 = icmp sge i32 %smax2, %a96 %smax_or_eq3 = select i1 %c3, i32 %smax2, i32 %a97 %res = add i32 %smax_or_eq1, %smax_or_eq398 ret i32 %res99}100 101; m1 = smax(a,b) ; has side uses102; m2 = smax(umax((b,c), a) ; check that signed and unsigned maxs are not mixed103define i32 @smax_test6(i32 %a, i32 %b, i32 %c) {104; CHECK-LABEL: @smax_test6(105; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[A:%.*]], [[B:%.*]]106; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]107; CHECK-NEXT: [[C2:%.*]] = icmp ugt i32 [[B]], [[C:%.*]]108; CHECK-NEXT: [[UMAX2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]109; CHECK-NEXT: [[C3:%.*]] = icmp sgt i32 [[UMAX2]], [[A]]110; CHECK-NEXT: [[SMAX3:%.*]] = select i1 [[C3]], i32 [[UMAX2]], i32 [[A]]111; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3]]112; CHECK-NEXT: ret i32 [[RES]]113;114 %c1 = icmp sgt i32 %a, %b115 %smax1 = select i1 %c1, i32 %a, i32 %b116 %c2 = icmp ugt i32 %b, %c117 %umax2 = select i1 %c2, i32 %b, i32 %c118 %c3 = icmp sgt i32 %umax2, %a119 %smax3 = select i1 %c3, i32 %umax2, i32 %a120 %res = add i32 %smax1, %smax3121 ret i32 %res122}123 124; m1 = smax(a,b) ; has side uses125; m2 = smax(smin((b,c), a) ; check that max and min are not mixed126define i32 @smax_test7(i32 %a, i32 %b, i32 %c) {127; CHECK-LABEL: @smax_test7(128; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[A:%.*]], [[B:%.*]]129; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]130; CHECK-NEXT: [[C2:%.*]] = icmp slt i32 [[B]], [[C:%.*]]131; CHECK-NEXT: [[SMIN2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]132; CHECK-NEXT: [[C3:%.*]] = icmp slt i32 [[SMIN2]], [[A]]133; CHECK-NEXT: [[SMAX3:%.*]] = select i1 [[C3]], i32 [[SMIN2]], i32 [[A]]134; CHECK-NEXT: [[RES:%.*]] = add i32 [[SMAX1]], [[SMAX3]]135; CHECK-NEXT: ret i32 [[RES]]136;137 %c1 = icmp sgt i32 %a, %b138 %smax1 = select i1 %c1, i32 %a, i32 %b139 %c2 = icmp slt i32 %b, %c140 %smin2 = select i1 %c2, i32 %b, i32 %c141 %c3 = icmp slt i32 %smin2, %a142 %smax3 = select i1 %c3, i32 %smin2, i32 %a143 %res = add i32 %smax1, %smax3144 ret i32 %res145}146 147; Pointer types are not supported yet148define ptr @smax_test8(ptr %a, ptr %b, ptr %c) {149; CHECK-LABEL: @smax_test8(150; CHECK-NEXT: [[C1:%.*]] = icmp sgt ptr [[A:%.*]], [[B:%.*]]151; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], ptr [[A]], ptr [[B]]152; CHECK-NEXT: [[C2:%.*]] = icmp sgt ptr [[B]], [[C:%.*]]153; CHECK-NEXT: [[SMAX2:%.*]] = select i1 [[C2]], ptr [[B]], ptr [[C]]154; CHECK-NEXT: [[C3:%.*]] = icmp sgt ptr [[SMAX2]], [[A]]155; CHECK-NEXT: [[SMAX3:%.*]] = select i1 [[C3]], ptr [[SMAX2]], ptr [[A]]156; CHECK-NEXT: [[C4:%.*]] = icmp sgt ptr [[SMAX1]], [[SMAX3]]157; CHECK-NEXT: [[RES:%.*]] = select i1 [[C4]], ptr [[SMAX1]], ptr [[SMAX3]]158; CHECK-NEXT: ret ptr [[RES]]159;160 %c1 = icmp sgt ptr %a, %b161 %smax1 = select i1 %c1, ptr %a, ptr %b162 %c2 = icmp sgt ptr %b, %c163 %smax2 = select i1 %c2, ptr %b, ptr %c164 %c3 = icmp sgt ptr %smax2, %a165 %smax3 = select i1 %c3, ptr %smax2, ptr %a166 %c4 = icmp sgt ptr %smax1, %smax3167 %res = select i1 %c4, ptr %smax1, ptr %smax3168 ret ptr %res169}170 171