[llvm] 3f36bdd - [PhaseOrdering] add test for sdiv with common factor; NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2022-10-10T08:11:53-04:00
New Revision: 3f36bdd2f48626a24ec7729b7c9b96d1ff873645

URL: https://github.com/llvm/llvm-project/commit/3f36bdd2f48626a24ec7729b7c9b96d1ff873645
DIFF: https://github.com/llvm/llvm-project/commit/3f36bdd2f48626a24ec7729b7c9b96d1ff873645.diff

LOG: [PhaseOrdering] add test for sdiv with common factor; NFC

issue #58137

Added: 
    

Modified: 
    llvm/test/Transforms/PhaseOrdering/reassociate-instcombine.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/PhaseOrdering/reassociate-instcombine.ll b/llvm/test/Transforms/PhaseOrdering/reassociate-instcombine.ll
index fb9e5d0f5694a..281d5c7ce0944 100644
--- a/llvm/test/Transforms/PhaseOrdering/reassociate-instcombine.ll
+++ b/llvm/test/Transforms/PhaseOrdering/reassociate-instcombine.ll
@@ -37,8 +37,8 @@ define i32 @not_reassociate_or_or_not(i32 %a, i32 %b, i32 %c, i32 %d) {
   ret i32 %b3
 }
 
-define i32 @PR58137(i32 %a, i32 %b) {
-; CHECK-LABEL: @PR58137(
+define i32 @PR58137_sdiv(i32 %a, i32 %b) {
+; CHECK-LABEL: @PR58137_sdiv(
 ; CHECK-NEXT:    ret i32 [[B:%.*]]
 ;
   %mul = mul nsw i32 2, %b
@@ -47,3 +47,24 @@ define i32 @PR58137(i32 %a, i32 %b) {
   %div = sdiv i32 %mul1, %mul2
   ret i32 %div
 }
+
+define i32 @PR58137_udiv(i32 %x, i32 %y) {
+; CHECK-LABEL: @PR58137_udiv(
+; CHECK-NEXT:    [[ZX:%.*]] = zext i32 [[X:%.*]] to i64
+; CHECK-NEXT:    [[ZY:%.*]] = zext i32 [[Y:%.*]] to i64
+; CHECK-NEXT:    [[M1:%.*]] = shl nuw nsw i64 [[ZX]], 2
+; CHECK-NEXT:    [[M2:%.*]] = mul i64 [[M1]], [[ZY]]
+; CHECK-NEXT:    [[M3:%.*]] = shl nuw nsw i64 [[ZY]], 2
+; CHECK-NEXT:    [[D:%.*]] = udiv i64 [[M2]], [[M3]]
+; CHECK-NEXT:    [[T:%.*]] = trunc i64 [[D]] to i32
+; CHECK-NEXT:    ret i32 [[T]]
+;
+  %zx = zext i32 %x to i64
+  %zy = zext i32 %y to i64
+  %m1 = mul nuw i64 %zx, %zy
+  %m2 = mul nuw i64 4, %m1
+  %m3 = mul nuw i64 4, %zy
+  %d = udiv i64 %m2, %m3
+  %t = trunc i64 %d to i32
+  ret i32 %t
+}