170 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.umax.i32(i32 %a, i32 %b)5 6; m1 = umax(a,b) ; has side uses7; m2 = umax(umax((b,c), a) -> m2 = umax(m1, c)8define i32 @umax_test1(i32 %a, i32 %b, i32 %c) {9; CHECK-LABEL: @umax_test1(10; CHECK-NEXT: [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]11; CHECK-NEXT: [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]12; CHECK-NEXT: [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])13; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]14; CHECK-NEXT: ret i32 [[RES]]15;16 %c1 = icmp ugt i32 %a, %b17 %umax1 = select i1 %c1, i32 %a, i32 %b18 %c2 = icmp ugt i32 %b, %c19 %umax2 = select i1 %c2, i32 %b, i32 %c20 %c3 = icmp ugt i32 %umax2, %a21 %umax3 = select i1 %c3, i32 %umax2, i32 %a22 %res = add i32 %umax1, %umax323 ret i32 %res24}25 26; m1 = umax(a,b) ; has side uses27; m2 = umax(b, (umax(a, c))) -> m2 = umax(m1, c)28define i32 @umax_test2(i32 %a, i32 %b, i32 %c) {29; CHECK-LABEL: @umax_test2(30; CHECK-NEXT: [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]31; CHECK-NEXT: [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]32; CHECK-NEXT: [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])33; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]34; CHECK-NEXT: ret i32 [[RES]]35;36 %c1 = icmp ugt i32 %a, %b37 %umax1 = select i1 %c1, i32 %a, i32 %b38 %c2 = icmp ugt i32 %a, %c39 %umax2 = select i1 %c2, i32 %a, i32 %c40 %c3 = icmp ugt i32 %b, %umax241 %umax3 = select i1 %c3, i32 %b, i32 %umax242 %res = add i32 %umax1, %umax343 ret i32 %res44}45 46; Same test as umax_test1 but uses @llvm.umax intrinsic47define i32 @umax_test3(i32 %a, i32 %b, i32 %c) {48; CHECK-LABEL: @umax_test3(49; CHECK-NEXT: [[UMAX1:%.*]] = call i32 @llvm.umax.i32(i32 [[A:%.*]], i32 [[B:%.*]])50; CHECK-NEXT: [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])51; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]52; CHECK-NEXT: ret i32 [[RES]]53;54 %umax1 = call i32 @llvm.umax.i32(i32 %a, i32 %b)55 %umax2 = call i32 @llvm.umax.i32(i32 %b, i32 %c)56 %umax3 = call i32 @llvm.umax.i32(i32 %umax2, i32 %a)57 %res = add i32 %umax1, %umax358 ret i32 %res59}60 61; m1 = umax(a,b) ; has side uses62; m2 = umax(umax_or_eq((b,c), a) -> m2 = umax(m1, c)63define i32 @umax_test4(i32 %a, i32 %b, i32 %c) {64; CHECK-LABEL: @umax_test4(65; CHECK-NEXT: [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]66; CHECK-NEXT: [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]67; CHECK-NEXT: [[UMAX3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX1]], i32 [[C:%.*]])68; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3_NARY]]69; CHECK-NEXT: ret i32 [[RES]]70;71 %c1 = icmp ugt i32 %a, %b72 %umax1 = select i1 %c1, i32 %a, i32 %b73 %c2 = icmp uge i32 %b, %c74 %umax_or_eq2 = select i1 %c2, i32 %b, i32 %c75 %c3 = icmp ugt i32 %umax_or_eq2, %a76 %umax3 = select i1 %c3, i32 %umax_or_eq2, i32 %a77 %res = add i32 %umax1, %umax378 ret i32 %res79}80 81; m1 = umax_or_eq(a,b) ; has side uses82; m2 = umax_or_eq(umax((b,c), a) -> m2 = umax(m1, c)83define i32 @umax_test5(i32 %a, i32 %b, i32 %c) {84; CHECK-LABEL: @umax_test5(85; CHECK-NEXT: [[C1:%.*]] = icmp uge i32 [[A:%.*]], [[B:%.*]]86; CHECK-NEXT: [[UMAX_OR_EQ1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]87; CHECK-NEXT: [[UMAX_OR_EQ3_NARY:%.*]] = call i32 @llvm.umax.i32(i32 [[UMAX_OR_EQ1]], i32 [[C:%.*]])88; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX_OR_EQ1]], [[UMAX_OR_EQ3_NARY]]89; CHECK-NEXT: ret i32 [[RES]]90;91 %c1 = icmp uge i32 %a, %b92 %umax_or_eq1 = select i1 %c1, i32 %a, i32 %b93 %c2 = icmp ugt i32 %b, %c94 %umax2 = select i1 %c2, i32 %b, i32 %c95 %c3 = icmp uge i32 %umax2, %a96 %umax_or_eq3 = select i1 %c3, i32 %umax2, i32 %a97 %res = add i32 %umax_or_eq1, %umax_or_eq398 ret i32 %res99}100 101; m1 = umax(a,b) ; has side uses102; m2 = umax(smax((b,c), a) ; check that signed and unsigned maxs are not mixed103define i32 @umax_test6(i32 %a, i32 %b, i32 %c) {104; CHECK-LABEL: @umax_test6(105; CHECK-NEXT: [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]106; CHECK-NEXT: [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]107; CHECK-NEXT: [[C2:%.*]] = icmp sgt i32 [[B]], [[C:%.*]]108; CHECK-NEXT: [[SMAX2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]109; CHECK-NEXT: [[C3:%.*]] = icmp ugt i32 [[SMAX2]], [[A]]110; CHECK-NEXT: [[UMAX3:%.*]] = select i1 [[C3]], i32 [[SMAX2]], i32 [[A]]111; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3]]112; CHECK-NEXT: ret i32 [[RES]]113;114 %c1 = icmp ugt i32 %a, %b115 %umax1 = select i1 %c1, i32 %a, i32 %b116 %c2 = icmp sgt i32 %b, %c117 %smax2 = select i1 %c2, i32 %b, i32 %c118 %c3 = icmp ugt i32 %smax2, %a119 %umax3 = select i1 %c3, i32 %smax2, i32 %a120 %res = add i32 %umax1, %umax3121 ret i32 %res122}123 124; m1 = umax(a,b) ; has side uses125; m2 = umax(umin((b,c), a) ; check that max and min are not mixed126define i32 @umax_test7(i32 %a, i32 %b, i32 %c) {127; CHECK-LABEL: @umax_test7(128; CHECK-NEXT: [[C1:%.*]] = icmp ugt i32 [[A:%.*]], [[B:%.*]]129; CHECK-NEXT: [[UMAX1:%.*]] = select i1 [[C1]], i32 [[A]], i32 [[B]]130; CHECK-NEXT: [[C2:%.*]] = icmp ult i32 [[B]], [[C:%.*]]131; CHECK-NEXT: [[UMAX2:%.*]] = select i1 [[C2]], i32 [[B]], i32 [[C]]132; CHECK-NEXT: [[C3:%.*]] = icmp ugt i32 [[UMAX2]], [[A]]133; CHECK-NEXT: [[UMAX3:%.*]] = select i1 [[C3]], i32 [[UMAX2]], i32 [[A]]134; CHECK-NEXT: [[RES:%.*]] = add i32 [[UMAX1]], [[UMAX3]]135; CHECK-NEXT: ret i32 [[RES]]136;137 %c1 = icmp ugt i32 %a, %b138 %umax1 = select i1 %c1, i32 %a, i32 %b139 %c2 = icmp ult i32 %b, %c140 %umax2 = select i1 %c2, i32 %b, i32 %c141 %c3 = icmp ugt i32 %umax2, %a142 %umax3 = select i1 %c3, i32 %umax2, i32 %a143 %res = add i32 %umax1, %umax3144 ret i32 %res145}146 147; Pointer types are not supported yet148define ptr @umax_test8(ptr %a, ptr %b, ptr %c) {149; CHECK-LABEL: @umax_test8(150; CHECK-NEXT: [[C1:%.*]] = icmp ugt ptr [[A:%.*]], [[B:%.*]]151; CHECK-NEXT: [[SMAX1:%.*]] = select i1 [[C1]], ptr [[A]], ptr [[B]]152; CHECK-NEXT: [[C2:%.*]] = icmp ugt ptr [[B]], [[C:%.*]]153; CHECK-NEXT: [[SMAX2:%.*]] = select i1 [[C2]], ptr [[B]], ptr [[C]]154; CHECK-NEXT: [[C3:%.*]] = icmp ugt ptr [[SMAX2]], [[A]]155; CHECK-NEXT: [[SMAX3:%.*]] = select i1 [[C3]], ptr [[SMAX2]], ptr [[A]]156; CHECK-NEXT: [[C4:%.*]] = icmp ugt ptr [[SMAX1]], [[SMAX3]]157; CHECK-NEXT: [[RES:%.*]] = select i1 [[C4]], ptr [[SMAX1]], ptr [[SMAX3]]158; CHECK-NEXT: ret ptr [[RES]]159;160 %c1 = icmp ugt ptr %a, %b161 %smax1 = select i1 %c1, ptr %a, ptr %b162 %c2 = icmp ugt ptr %b, %c163 %smax2 = select i1 %c2, ptr %b, ptr %c164 %c3 = icmp ugt ptr %smax2, %a165 %smax3 = select i1 %c3, ptr %smax2, ptr %a166 %c4 = icmp ugt ptr %smax1, %smax3167 %res = select i1 %c4, ptr %smax1, ptr %smax3168 ret ptr %res169}170