[ConstantFolding] add tests for funnel shift intrinsics; NFC

No functionality for this yet.

llvm-svn: 339889

1 files changed, 89 insertions, 0 deletions

diff --git a/llvm/test/Analysis/ConstantFolding/funnel-shift.ll b/llvm/test/Analysis/ConstantFolding/funnel-shift.ll
new file mode 100644
index 00000000000..86a4d62a06e
--- /dev/null
+++ b/llvm/test/Analysis/ConstantFolding/funnel-shift.ll
@@ -0,0 +1,89 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -constprop -S -o - | FileCheck %s
+
+declare i32 @llvm.fshl.i32(i32, i32, i32)
+declare i32 @llvm.fshr.i32(i32, i32, i32)
+declare i7 @llvm.fshl.i7(i7, i7, i7)
+declare i7 @llvm.fshr.i7(i7, i7, i7)
+declare <4 x i8> @llvm.fshl.v4i8(<4 x i8>, <4 x i8>, <4 x i8>)
+declare <4 x i8> @llvm.fshr.v4i8(<4 x i8>, <4 x i8>, <4 x i8>)
+
+; extract(concat(0x12345678, 0xABCDEF01) << 5) = 0x468ACF15
+
+define i32 @fshl_i32() {
+; CHECK-LABEL: @fshl_i32(
+; CHECK-NEXT:    [[F:%.*]] = call i32 @llvm.fshl.i32(i32 305419896, i32 -1412567295, i32 5)
+; CHECK-NEXT:    ret i32 [[F]]
+;
+  %f = call i32 @llvm.fshl.i32(i32 305419896, i32 2882400001, i32 5)
+  ret i32 %f
+}
+
+; extract(concat(0x12345678, 0xABCDEF01) >> 5) = 0xC55E6F78
+; Try an oversized shift to test modulo functionality.
+
+define i32 @fshr_i32() {
+; CHECK-LABEL: @fshr_i32(
+; CHECK-NEXT:    [[F:%.*]] = call i32 @llvm.fshr.i32(i32 305419896, i32 -1412567295, i32 37)
+; CHECK-NEXT:    ret i32 [[F]]
+;
+  %f = call i32 @llvm.fshr.i32(i32 305419896, i32 2882400001, i32 37)
+  ret i32 %f
+}
+
+; Use a weird type.
+; Try an oversized shift to test modulo functionality.
+
+; extract(concat(0b1110000, 0b1111111) << 2) = 0b1000011
+
+define i7 @fshl_i7() {
+; CHECK-LABEL: @fshl_i7(
+; CHECK-NEXT:    [[F:%.*]] = call i7 @llvm.fshl.i7(i7 -16, i7 -1, i7 9)
+; CHECK-NEXT:    ret i7 [[F]]
+;
+  %f = call i7 @llvm.fshl.i7(i7 112, i7 127, i7 9)
+  ret i7 %f
+}
+
+; extract(concat(0b1110000, 0b1111111) >> 2) = 0b0011111
+; Try an oversized shift to test modulo functionality.
+
+define i7 @fshr_i7() {
+; CHECK-LABEL: @fshr_i7(
+; CHECK-NEXT:    [[F:%.*]] = call i7 @llvm.fshr.i7(i7 -16, i7 -1, i7 16)
+; CHECK-NEXT:    ret i7 [[F]]
+;
+  %f = call i7 @llvm.fshr.i7(i7 112, i7 127, i7 16)
+  ret i7 %f
+}
+
+; Vectors are folded by handling each scalar element individually, so this is the equivalent of 4 scalar tests:
+; extract(concat(0x00, 0xFF) << 0) = 0x00
+; extract(concat(0xFF, 0x00) << 0) = 0xFF
+; extract(concat(0x10, 0x55) << 1) = 0x20
+; extract(concat(0x11, 0xAA) << 2) = 0x46
+
+define <4 x i8> @fshl_v4i8() {
+; CHECK-LABEL: @fshl_v4i8(
+; CHECK-NEXT:    [[F:%.*]] = call <4 x i8> @llvm.fshl.v4i8(<4 x i8> <i8 0, i8 -1, i8 16, i8 17>, <4 x i8> <i8 -1, i8 0, i8 85, i8 -86>, <4 x i8> <i8 0, i8 8, i8 9, i8 10>)
+; CHECK-NEXT:    ret <4 x i8> [[F]]
+;
+  %f = call <4 x i8> @llvm.fshl.v4i8(<4 x i8> <i8 0, i8 -1, i8 16, i8 17>, <4 x i8> <i8 -1, i8 0, i8 85, i8 170>, <4 x i8> <i8 0, i8 8, i8 9, i8 10>)
+  ret <4 x i8> %f
+}
+
+; Vectors are folded by handling each scalar element individually, so this is the equivalent of 4 scalar tests:
+; extract(concat(0x00, 0xFF) >> 0) = 0xFF
+; extract(concat(0xFF, 0x00) >> 0) = 0x00
+; extract(concat(0x10, 0x55) >> 1) = 0x2A
+; extract(concat(0x11, 0xAA) >> 2) = 0x6A
+
+define <4 x i8> @fshr_v4i8() {
+; CHECK-LABEL: @fshr_v4i8(
+; CHECK-NEXT:    [[F:%.*]] = call <4 x i8> @llvm.fshr.v4i8(<4 x i8> <i8 0, i8 -1, i8 16, i8 17>, <4 x i8> <i8 -1, i8 0, i8 85, i8 -86>, <4 x i8> <i8 0, i8 8, i8 9, i8 10>)
+; CHECK-NEXT:    ret <4 x i8> [[F]]
+;
+  %f = call <4 x i8> @llvm.fshr.v4i8(<4 x i8> <i8 0, i8 -1, i8 16, i8 17>, <4 x i8> <i8 -1, i8 0, i8 85, i8 170>, <4 x i8> <i8 0, i8 8, i8 9, i8 10>)
+  ret <4 x i8> %f
+}
+