[llvm] fc9d54a - [InstCombine] add tests for smul/umul with overflow with i1 types; NFC
Sanjay Patel via llvm-commits
llvm-commits at lists.llvm.org
Mon Jan 9 07:28:52 PST 2023
Author: Sanjay Patel
Date: 2023-01-09T10:27:15-05:00
New Revision: fc9d54a4a131cbbd51c135b38b590c36357c8aa2
URL: https://github.com/llvm/llvm-project/commit/fc9d54a4a131cbbd51c135b38b590c36357c8aa2
DIFF: https://github.com/llvm/llvm-project/commit/fc9d54a4a131cbbd51c135b38b590c36357c8aa2.diff
LOG: [InstCombine] add tests for smul/umul with overflow with i1 types; NFC
More coverage for D141214 / issue #59876
Added:
Modified:
llvm/test/Transforms/InstCombine/smulo.ll
llvm/test/Transforms/InstCombine/umulo.ll
Removed:
################################################################################
diff --git a/llvm/test/Transforms/InstCombine/smulo.ll b/llvm/test/Transforms/InstCombine/smulo.ll
index 132c2c5c63a2..6f88e201fae9 100644
--- a/llvm/test/Transforms/InstCombine/smulo.ll
+++ b/llvm/test/Transforms/InstCombine/smulo.ll
@@ -3,6 +3,8 @@
declare { i64, i1 } @llvm.smul.with.overflow.i64(i64, i64)
declare { i8, i1 } @llvm.smul.with.overflow.i8(i8, i8)
+declare { i1, i1 } @llvm.smul.with.overflow.i1(i1, i1)
+declare { <2 x i1>, <2 x i1> } @llvm.smul.with.overflow.v2i1(<2 x i1>, <2 x i1>)
define i1 @test_generic(i64 %a, i64 %b) {
; CHECK-LABEL: @test_generic(
@@ -98,3 +100,80 @@ define i1 @test_constant255(i8 %a) {
ret i1 %overflow
}
+define i1 @i1_res(i1 %x, i1 %y) {
+; CHECK-LABEL: @i1_res(
+; CHECK-NEXT: [[R:%.*]] = and i1 [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT: ret i1 [[R]]
+;
+ %m = call {i1, i1} @llvm.smul.with.overflow.i1(i1 %x, i1 %y)
+ %r = extractvalue {i1, i1} %m, 0
+ ret i1 %r
+}
+
+define <2 x i1> @v2i1_res(<2 x i1> %x, <2 x i1> %y) {
+; CHECK-LABEL: @v2i1_res(
+; CHECK-NEXT: [[R:%.*]] = and <2 x i1> [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT: ret <2 x i1> [[R]]
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.smul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> %y)
+ %r = extractvalue {<2 x i1>, <2 x i1>} %m, 0
+ ret <2 x i1> %r
+}
+
+define i1 @i1_res_by_one(i1 %x) {
+; CHECK-LABEL: @i1_res_by_one(
+; CHECK-NEXT: ret i1 [[X:%.*]]
+;
+ %m = call {i1, i1} @llvm.smul.with.overflow.i1(i1 %x, i1 1)
+ %r = extractvalue {i1, i1} %m, 0
+ ret i1 %r
+}
+
+define <2 x i1> @v2i1_res_by_one(<2 x i1> %x) {
+; CHECK-LABEL: @v2i1_res_by_one(
+; CHECK-NEXT: ret <2 x i1> [[X:%.*]]
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.smul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> <i1 1, i1 1>)
+ %r = extractvalue {<2 x i1>, <2 x i1>} %m, 0
+ ret <2 x i1> %r
+}
+
+define i1 @i1_ov(i1 %x, i1 %y) {
+; CHECK-LABEL: @i1_ov(
+; CHECK-NEXT: [[M:%.*]] = call { i1, i1 } @llvm.smul.with.overflow.i1(i1 [[X:%.*]], i1 [[Y:%.*]])
+; CHECK-NEXT: [[OV:%.*]] = extractvalue { i1, i1 } [[M]], 1
+; CHECK-NEXT: ret i1 [[OV]]
+;
+ %m = call {i1, i1} @llvm.smul.with.overflow.i1(i1 %x, i1 %y)
+ %ov = extractvalue {i1, i1} %m, 1
+ ret i1 %ov
+}
+
+define <2 x i1> @v2i1_ov(<2 x i1> %x, <2 x i1> %y) {
+; CHECK-LABEL: @v2i1_ov(
+; CHECK-NEXT: [[M:%.*]] = call { <2 x i1>, <2 x i1> } @llvm.smul.with.overflow.v2i1(<2 x i1> [[X:%.*]], <2 x i1> [[Y:%.*]])
+; CHECK-NEXT: [[OV:%.*]] = extractvalue { <2 x i1>, <2 x i1> } [[M]], 1
+; CHECK-NEXT: ret <2 x i1> [[OV]]
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.smul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> %y)
+ %ov = extractvalue {<2 x i1>, <2 x i1>} %m, 1
+ ret <2 x i1> %ov
+}
+
+define i1 @i1_ov_by_one(i1 %x) {
+; CHECK-LABEL: @i1_ov_by_one(
+; CHECK-NEXT: ret i1 false
+;
+ %m = call {i1, i1} @llvm.smul.with.overflow.i1(i1 %x, i1 1)
+ %ov = extractvalue {i1, i1} %m, 1
+ ret i1 %ov
+}
+
+define <2 x i1> @v2i1_ov_by_one(<2 x i1> %x) {
+; CHECK-LABEL: @v2i1_ov_by_one(
+; CHECK-NEXT: ret <2 x i1> zeroinitializer
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.smul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> <i1 1, i1 1>)
+ %ov = extractvalue {<2 x i1>, <2 x i1>} %m, 1
+ ret <2 x i1> %ov
+}
diff --git a/llvm/test/Transforms/InstCombine/umulo.ll b/llvm/test/Transforms/InstCombine/umulo.ll
index 8a5747da2520..4e5c1feb7a70 100644
--- a/llvm/test/Transforms/InstCombine/umulo.ll
+++ b/llvm/test/Transforms/InstCombine/umulo.ll
@@ -3,6 +3,8 @@
declare { i64, i1 } @llvm.umul.with.overflow.i64(i64, i64)
declare { i8, i1 } @llvm.umul.with.overflow.i8(i8, i8)
+declare { i1, i1 } @llvm.umul.with.overflow.i1(i1, i1)
+declare { <2 x i1>, <2 x i1> } @llvm.umul.with.overflow.v2i1(<2 x i1>, <2 x i1>)
define i1 @test_generic(i64 %a, i64 %b) {
; CHECK-LABEL: @test_generic(
@@ -94,3 +96,76 @@ define i1 @test_constant255(i8 %a) {
ret i1 %overflow
}
+define i1 @i1_res(i1 %x, i1 %y) {
+; CHECK-LABEL: @i1_res(
+; CHECK-NEXT: [[M:%.*]] = and i1 [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT: ret i1 [[M]]
+;
+ %m = call {i1, i1} @llvm.umul.with.overflow.i1(i1 %x, i1 %y)
+ %r = extractvalue {i1, i1} %m, 0
+ ret i1 %r
+}
+
+define <2 x i1> @v2i1_res(<2 x i1> %x, <2 x i1> %y) {
+; CHECK-LABEL: @v2i1_res(
+; CHECK-NEXT: [[M:%.*]] = and <2 x i1> [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT: ret <2 x i1> [[M]]
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.umul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> %y)
+ %r = extractvalue {<2 x i1>, <2 x i1>} %m, 0
+ ret <2 x i1> %r
+}
+
+define i1 @i1_res_by_one(i1 %x) {
+; CHECK-LABEL: @i1_res_by_one(
+; CHECK-NEXT: ret i1 [[X:%.*]]
+;
+ %m = call {i1, i1} @llvm.umul.with.overflow.i1(i1 %x, i1 1)
+ %r = extractvalue {i1, i1} %m, 0
+ ret i1 %r
+}
+
+define <2 x i1> @v2i1_res_by_one(<2 x i1> %x) {
+; CHECK-LABEL: @v2i1_res_by_one(
+; CHECK-NEXT: ret <2 x i1> [[X:%.*]]
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.umul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> <i1 1, i1 1>)
+ %r = extractvalue {<2 x i1>, <2 x i1>} %m, 0
+ ret <2 x i1> %r
+}
+
+define i1 @i1_ov(i1 %x, i1 %y) {
+; CHECK-LABEL: @i1_ov(
+; CHECK-NEXT: ret i1 false
+;
+ %m = call {i1, i1} @llvm.umul.with.overflow.i1(i1 %x, i1 %y)
+ %ov = extractvalue {i1, i1} %m, 1
+ ret i1 %ov
+}
+
+define <2 x i1> @v2i1_ov(<2 x i1> %x, <2 x i1> %y) {
+; CHECK-LABEL: @v2i1_ov(
+; CHECK-NEXT: ret <2 x i1> zeroinitializer
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.umul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> %y)
+ %ov = extractvalue {<2 x i1>, <2 x i1>} %m, 1
+ ret <2 x i1> %ov
+}
+
+define i1 @i1_ov_by_one(i1 %x) {
+; CHECK-LABEL: @i1_ov_by_one(
+; CHECK-NEXT: ret i1 false
+;
+ %m = call {i1, i1} @llvm.umul.with.overflow.i1(i1 %x, i1 1)
+ %ov = extractvalue {i1, i1} %m, 1
+ ret i1 %ov
+}
+
+define <2 x i1> @v2i1_ov_by_one(<2 x i1> %x) {
+; CHECK-LABEL: @v2i1_ov_by_one(
+; CHECK-NEXT: ret <2 x i1> zeroinitializer
+;
+ %m = call {<2 x i1>, <2 x i1>} @llvm.umul.with.overflow.v2i1(<2 x i1> %x, <2 x i1> <i1 1, i1 1>)
+ %ov = extractvalue {<2 x i1>, <2 x i1>} %m, 1
+ ret <2 x i1> %ov
+}
More information about the llvm-commits
mailing list