chore(InstCombine): add more types to the test

Author: bassiounix

llvm/test/Transforms/InstCombine/iX-ext-split.ll

@@ -3,6 +3,7 @@
 
 ; PR129363
 
+; ext split from i32 to i128
 define i128 @i128_ext_split(i32 noundef %x) {
 ; CHECK-LABEL: define i128 @i128_ext_split(
 ; CHECK-SAME: i32 noundef [[X:%.*]]) {
@@ -19,6 +20,7 @@ define i128 @i128_ext_split(i32 noundef %x) {
   ret i128 %x.sroa.0.0.insert.insert.i
 }
 
+; ext split from i32 to i128
 define void @i128_ext_split_store(i32 %x, ptr %out)  {
 ; CHECK-LABEL: define void @i128_ext_split_store(
 ; CHECK-SAME: i32 [[X:%.*]], ptr [[OUT:%.*]]) {
@@ -40,3 +42,49 @@ entry:
   store i128 %res, ptr %out, align 16
   ret void
 }
+
+; ext split from i16 to i64
+define void @i64_ext_split_store(i16 %x, ptr %out)  {
+; CHECK-LABEL: define void @i64_ext_split_store(
+; CHECK-SAME: i16 [[X:%.*]], ptr [[OUT:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    [[RES:%.*]] = sext i16 [[X]] to i64
+; CHECK-NEXT:    store i64 [[RES]], ptr [[OUT]], align 16
+; CHECK-NEXT:    ret void
+;
+entry:
+  %LowerSrc = sext i16 %x to i32
+  %lo = zext i32 %LowerSrc to i64
+
+  %sign = ashr i16 %x, 15
+  %UpperSrc = sext i16 %sign to i32
+  %widen = zext i32 %UpperSrc to i64
+  %hi = shl nuw i64 %widen, 32
+
+  %res = or disjoint i64 %hi, %lo
+  store i64 %res, ptr %out, align 16
+  ret void
+}
+
+; ext split from i16 to i128
+define void @i128_ext_split_store_i16(i16 %x, ptr %out)  {
+; CHECK-LABEL: define void @i128_ext_split_store_i16(
+; CHECK-SAME: i16 [[X:%.*]], ptr [[OUT:%.*]]) {
+; CHECK-NEXT:  [[ENTRY:.*:]]
+; CHECK-NEXT:    [[RES:%.*]] = sext i16 [[X]] to i128
+; CHECK-NEXT:    store i128 [[RES]], ptr [[OUT]], align 16
+; CHECK-NEXT:    ret void
+;
+entry:
+  %LowerSrc = sext i16 %x to i64
+  %lo = zext i64 %LowerSrc to i128
+
+  %sign = ashr i16 %x, 15
+  %UpperSrc = sext i16 %sign to i64
+  %widen = zext i64 %UpperSrc to i128
+  %hi = shl nuw i128 %widen, 64
+
+  %res = or disjoint i128 %hi, %lo
+  store i128 %res, ptr %out, align 16
+  ret void
+}