update test cases

Author: hassnaaHamdi

llvm/lib/CodeGen/CodeGenPrepare.cpp

@@ -6504,7 +6504,16 @@ bool CodeGenPrepare::optimizeMulWithOverflow(Instruction *I, bool IsSigned,
 
   // BB overflow.no:
   Builder.SetInsertPoint(NoOverflowBB);
-  Value *Mul = Builder.CreateMul(LHS, RHS, "mul.overflow.no");
+    Value *ExtLoLHS, *ExtLoRHS;
+  if (IsSigned) {
+    ExtLoLHS = Builder.CreateSExt(LoLHS, Ty, "lo.lhs.ext");
+    ExtLoRHS = Builder.CreateSExt(LoRHS, Ty, "lo.rhs.ext");
+  } else {
+    ExtLoLHS = Builder.CreateZExt(LoLHS, Ty, "lo.lhs.ext");
+    ExtLoRHS = Builder.CreateZExt(LoRHS, Ty, "lo.rhs.ext");
+  }
+
+  Value *Mul = Builder.CreateMul(ExtLoLHS, ExtLoRHS, "mul.overflow.no");
 
   // In overflow.no BB: we are sure that the overflow flag is false.
   // So, if we found this pattern:

llvm/test/CodeGen/AArch64/i128-math.ll

@@ -261,7 +261,7 @@ define i128 @u128_mul(i128 %x, i128 %y) {
 
 define { i128, i8 } @u128_checked_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: u128_checked_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    orr x8, x1, x3
 ; CHECK-NEXT:    cbz x8, .LBB17_2
 ; CHECK-NEXT:  // %bb.1: // %overflow
@@ -297,7 +297,7 @@ define { i128, i8 } @u128_checked_mul(i128 %x, i128 %y) {
 
 define { i128, i8 } @u128_overflowing_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: u128_overflowing_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    orr x8, x1, x3
 ; CHECK-NEXT:    cbz x8, .LBB18_2
 ; CHECK-NEXT:  // %bb.1: // %overflow
@@ -331,7 +331,7 @@ define { i128, i8 } @u128_overflowing_mul(i128 %x, i128 %y) {
 
 define i128 @u128_saturating_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: u128_saturating_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    orr x8, x1, x3
 ; CHECK-NEXT:    cbz x8, .LBB19_2
 ; CHECK-NEXT:  // %bb.1: // %overflow
@@ -379,7 +379,7 @@ define i128 @i128_mul(i128 %x, i128 %y) {
 
 define { i128, i8 } @i128_checked_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: i128_checked_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    eor x8, x3, x2, asr #63
 ; CHECK-NEXT:    eor x9, x1, x0, asr #63
 ; CHECK-NEXT:    orr x8, x9, x8
@@ -431,7 +431,7 @@ define { i128, i8 } @i128_checked_mul(i128 %x, i128 %y) {
 
 define { i128, i8 } @i128_overflowing_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: i128_overflowing_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    eor x8, x3, x2, asr #63
 ; CHECK-NEXT:    eor x9, x1, x0, asr #63
 ; CHECK-NEXT:    orr x8, x9, x8
@@ -481,7 +481,7 @@ define { i128, i8 } @i128_overflowing_mul(i128 %x, i128 %y) {
 
 define i128 @i128_saturating_mul(i128 %x, i128 %y) {
 ; CHECK-LABEL: i128_saturating_mul:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0:
 ; CHECK-NEXT:    eor x8, x3, x2, asr #63
 ; CHECK-NEXT:    eor x9, x1, x0, asr #63
 ; CHECK-NEXT:    orr x8, x9, x8

llvm/test/CodeGen/AArch64/i128_with_overflow.ll

@@ -223,7 +223,7 @@ cleanup:
 
 define i128 @test_umul_i128(i128 noundef %x, i128 noundef %y) {
 ; CHECK-LABEL: test_umul_i128:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0: // %entry
 ; CHECK-NEXT:    orr x8, x1, x3
 ; CHECK-NEXT:    cbz x8, .LBB4_3
 ; CHECK-NEXT:  // %bb.1: // %overflow
@@ -277,7 +277,7 @@ cleanup:
 
 define i128 @test_smul_i128(i128 noundef %x, i128 noundef %y) {
 ; CHECK-LABEL: test_smul_i128:
-; CHECK:       // %bb.0: // %overflow.entry
+; CHECK:       // %bb.0: // %entry
 ; CHECK-NEXT:    eor x8, x3, x2, asr #63
 ; CHECK-NEXT:    eor x9, x1, x0, asr #63
 ; CHECK-NEXT:    orr x8, x9, x8

llvm/test/CodeGen/AArch64/umulo-128-legalisation-lowering.ll

@@ -3,7 +3,7 @@
 
 define { i128, i8 } @muloti_test(i128 %l, i128 %r) unnamed_addr #0 {
 ; AARCH-LABEL: muloti_test:
-; AARCH:       // %bb.0: // %overflow.entry
+; AARCH:       // %bb.0: // %start
 ; AARCH-NEXT:    orr x8, x1, x3
 ; AARCH-NEXT:    cbz x8, .LBB0_2
 ; AARCH-NEXT:  // %bb.1: // %overflow
@@ -42,7 +42,7 @@ start:
 
 define i128 @__muloti4(i128 %0, i128 %1, ptr nocapture nonnull writeonly align 4 %2) #2 {
 ; AARCH-LABEL: __muloti4:
-; AARCH:       // %bb.0: // %overflow.entry
+; AARCH:       // %bb.0: // %Entry
 ; AARCH-NEXT:    eor x8, x3, x2, asr #63
 ; AARCH-NEXT:    eor x9, x1, x0, asr #63
 ; AARCH-NEXT:    str wzr, [x4]