[llvm] c1b4d7f - [InstCombine] Add shuffle-binop tests
Matt Devereau via llvm-commits
llvm-commits at lists.llvm.org
Fri Oct 28 09:02:24 PDT 2022
Author: Matt Devereau
Date: 2022-10-28T16:02:06Z
New Revision: c1b4d7f091273b1542b3e2539dae60f3f5eb4d02
URL: https://github.com/llvm/llvm-project/commit/c1b4d7f091273b1542b3e2539dae60f3f5eb4d02
DIFF: https://github.com/llvm/llvm-project/commit/c1b4d7f091273b1542b3e2539dae60f3f5eb4d02.diff
LOG: [InstCombine] Add shuffle-binop tests
These tests are a precommit for https://reviews.llvm.org/D135876
Added:
llvm/test/Transforms/InstCombine/shuffle-binop.ll
Modified:
Removed:
################################################################################
diff --git a/llvm/test/Transforms/InstCombine/shuffle-binop.ll b/llvm/test/Transforms/InstCombine/shuffle-binop.ll
new file mode 100644
index 0000000000000..fe2d1af5a04f3
--- /dev/null
+++ b/llvm/test/Transforms/InstCombine/shuffle-binop.ll
@@ -0,0 +1,157 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt -passes='instcombine' -S %s | FileCheck %s
+
+define <4 x i8> @splat_binop_non_splat_x(<4 x i8> %x, <4 x i8> %y) {
+; CHECK-LABEL: @splat_binop_non_splat_x(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <4 x i8> [[X:%.*]], <4 x i8> poison, <4 x i32> <i32 0, i32 2, i32 undef, i32 undef>
+; CHECK-NEXT: call void @use(<4 x i8> [[XSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = add <4 x i8> [[XSPLAT]], [[Y:%.*]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i8> [[B]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: ret <4 x i8> [[BSPLAT]]
+;
+ %xsplat = shufflevector <4 x i8> %x, <4 x i8> poison, <4 x i32> <i32 0, i32 2, i32 4, i32 6>
+ call void @use(<4 x i8> %xsplat)
+ %b = add <4 x i8> %xsplat, %y
+ %bsplat = shufflevector <4 x i8> %b, <4 x i8> poison, <4 x i32> zeroinitializer
+ ret <4 x i8> %bsplat
+}
+
+define <4 x i8> @non_splat_binop_splat_x(<4 x i8> %x, <4 x i8> %y) {
+; CHECK-LABEL: @non_splat_binop_splat_x(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <4 x i8> [[X:%.*]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i8> [[XSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = sub <4 x i8> [[XSPLAT]], [[Y:%.*]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i8> [[B]], <4 x i8> poison, <4 x i32> <i32 0, i32 2, i32 undef, i32 undef>
+; CHECK-NEXT: ret <4 x i8> [[BSPLAT]]
+;
+ %xsplat = shufflevector <4 x i8> %x, <4 x i8> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i8> %xsplat)
+ %b = sub <4 x i8> %xsplat, %y
+ %bsplat = shufflevector <4 x i8> %b, <4 x i8> poison, <4 x i32> <i32 0, i32 2, i32 4, i32 6>
+ ret <4 x i8> %bsplat
+}
+
+define <4 x i32> @splat_binop_splat_changes_x_length(<8 x i32> %x, <4 x i32> %y) {
+; CHECK-LABEL: @splat_binop_splat_changes_x_length(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <8 x i32> [[X:%.*]], <8 x i32> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i32> [[XSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = mul <4 x i32> [[XSPLAT]], [[Y:%.*]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i32> [[B]], <4 x i32> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: ret <4 x i32> [[BSPLAT]]
+;
+ %xsplat = shufflevector <8 x i32> %x, <8 x i32> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i32> %xsplat)
+ %b = mul <4 x i32> %xsplat, %y
+ %bsplat = shufflevector <4 x i32> %b, <4 x i32> poison, <4 x i32> zeroinitializer
+ ret <4 x i32> %bsplat
+}
+
+define <4 x i8> @splat_binop_splat_x(<4 x i8> %x, <4 x i8> %y) {
+; CHECK-LABEL: @splat_binop_splat_x(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <4 x i8> [[X:%.*]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i8> [[XSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = add <4 x i8> [[XSPLAT]], [[Y:%.*]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i8> [[B]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: ret <4 x i8> [[BSPLAT]]
+;
+ %xsplat = shufflevector <4 x i8> %x, <4 x i8> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i8> %xsplat)
+ %b = add <4 x i8> %xsplat, %y
+ %bsplat = shufflevector <4 x i8> %b, <4 x i8> poison, <4 x i32> zeroinitializer
+ ret <4 x i8> %bsplat
+}
+
+define <4 x i8> @splat_binop_splat_y(<4 x i8> %x, <4 x i8> %y) {
+; CHECK-LABEL: @splat_binop_splat_y(
+; CHECK-NEXT: [[YSPLAT:%.*]] = shufflevector <4 x i8> [[Y:%.*]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i8> [[YSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = sub <4 x i8> [[X:%.*]], [[YSPLAT]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i8> [[B]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: ret <4 x i8> [[BSPLAT]]
+;
+ %ysplat = shufflevector <4 x i8> %y, <4 x i8> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i8> %ysplat)
+ %b = sub <4 x i8> %x, %ysplat
+ %bsplat = shufflevector <4 x i8> %b, <4 x i8> poison, <4 x i32> zeroinitializer
+ ret <4 x i8> %bsplat
+}
+
+define <4 x i8> @splat_binop_splat_x_splat_y(<4 x i8> %x, <4 x i8> %y) {
+; CHECK-LABEL: @splat_binop_splat_x_splat_y(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <4 x i8> [[X:%.*]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i8> [[XSPLAT]])
+; CHECK-NEXT: [[YSPLAT:%.*]] = shufflevector <4 x i8> [[Y:%.*]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use(<4 x i8> [[YSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = mul <4 x i8> [[XSPLAT]], [[YSPLAT]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <4 x i8> [[B]], <4 x i8> poison, <4 x i32> zeroinitializer
+; CHECK-NEXT: ret <4 x i8> [[BSPLAT]]
+;
+ %xsplat = shufflevector <4 x i8> %x, <4 x i8> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i8> %xsplat)
+ %ysplat = shufflevector <4 x i8> %y, <4 x i8> poison, <4 x i32> zeroinitializer
+ call void @use(<4 x i8> %ysplat)
+ %b = mul <4 x i8> %xsplat, %ysplat
+ %bsplat = shufflevector <4 x i8> %b, <4 x i8> poison, <4 x i32> zeroinitializer
+ ret <4 x i8> %bsplat
+}
+
+define <vscale x 4 x i32> @vscale_splat_binop_splat_x(<vscale x 4 x i32> %x, <vscale x 4 x i32> %y) {
+; CHECK-LABEL: @vscale_splat_binop_splat_x(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[X:%.*]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: [[B:%.*]] = udiv <vscale x 4 x i32> [[XSPLAT]], [[Y:%.*]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[B]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: ret <vscale x 4 x i32> [[BSPLAT]]
+;
+ %xsplat = shufflevector <vscale x 4 x i32> %x, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ %b = udiv <vscale x 4 x i32> %xsplat, %y
+ %bsplat = shufflevector <vscale x 4 x i32> %b, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ ret <vscale x 4 x i32> %bsplat
+}
+
+define <vscale x 4 x i32> @vscale_splat_binop_splat_y(<vscale x 4 x i32> %x, <vscale x 4 x i32> %y) {
+; CHECK-LABEL: @vscale_splat_binop_splat_y(
+; CHECK-NEXT: [[YSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[Y:%.*]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: [[B:%.*]] = sdiv <vscale x 4 x i32> [[X:%.*]], [[YSPLAT]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[B]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: ret <vscale x 4 x i32> [[BSPLAT]]
+;
+ %ysplat = shufflevector <vscale x 4 x i32> %y, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ %b = sdiv <vscale x 4 x i32> %x, %ysplat
+ %bsplat = shufflevector <vscale x 4 x i32> %b, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ ret <vscale x 4 x i32> %bsplat
+}
+
+define <vscale x 4 x i32> @vscale_splat_binop_splat_x_splat_y(<vscale x 4 x i32> %x, <vscale x 4 x i32> %y) {
+; CHECK-LABEL: @vscale_splat_binop_splat_x_splat_y(
+; CHECK-NEXT: [[TMP1:%.*]] = ashr <vscale x 4 x i32> [[X:%.*]], [[Y:%.*]]
+; CHECK-NEXT: [[B:%.*]] = shufflevector <vscale x 4 x i32> [[TMP1]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: ret <vscale x 4 x i32> [[B]]
+;
+ %xsplat = shufflevector <vscale x 4 x i32> %x, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ %ysplat = shufflevector <vscale x 4 x i32> %y, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ %b = ashr <vscale x 4 x i32> %xsplat, %ysplat
+ %bsplat = shufflevector <vscale x 4 x i32> %b, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ ret <vscale x 4 x i32> %bsplat
+}
+
+define <vscale x 4 x i32> @vscale_splat_binop_splat_x_splat_y_calls(<vscale x 4 x i32> %x, <vscale x 4 x i32> %y) {
+; CHECK-LABEL: @vscale_splat_binop_splat_x_splat_y_calls(
+; CHECK-NEXT: [[XSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[X:%.*]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use_v(<vscale x 4 x i32> [[XSPLAT]])
+; CHECK-NEXT: [[YSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[Y:%.*]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: call void @use_v(<vscale x 4 x i32> [[YSPLAT]])
+; CHECK-NEXT: [[B:%.*]] = lshr <vscale x 4 x i32> [[XSPLAT]], [[YSPLAT]]
+; CHECK-NEXT: [[BSPLAT:%.*]] = shufflevector <vscale x 4 x i32> [[B]], <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+; CHECK-NEXT: ret <vscale x 4 x i32> [[BSPLAT]]
+;
+ %xsplat = shufflevector <vscale x 4 x i32> %x, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ call void @use_v(<vscale x 4 x i32> %xsplat)
+ %ysplat = shufflevector <vscale x 4 x i32> %y, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ call void @use_v(<vscale x 4 x i32> %ysplat)
+ %b = lshr <vscale x 4 x i32> %xsplat, %ysplat
+ %bsplat = shufflevector <vscale x 4 x i32> %b, <vscale x 4 x i32> poison, <vscale x 4 x i32> zeroinitializer
+ ret <vscale x 4 x i32> %bsplat
+}
+
+declare void @use(<4 x i8>)
+declare void @use_v(<vscale x 4 x i32>)
\ No newline at end of file
More information about the llvm-commits
mailing list