From 67a6b1c40c04178b52a53623e05a790f517622f1 Mon Sep 17 00:00:00 2001 From: Richard Sandiford Date: Thu, 11 Jul 2013 09:10:38 +0000 Subject: [PATCH] [SystemZ] Add testcase missing from r186073 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@186074 91177308-0d34-0410-b5e6-96231b3b80d8 --- test/CodeGen/SystemZ/risbg-01.ll | 266 +++++++++++++++++++++++++++++++ 1 file changed, 266 insertions(+) create mode 100644 test/CodeGen/SystemZ/risbg-01.ll diff --git a/test/CodeGen/SystemZ/risbg-01.ll b/test/CodeGen/SystemZ/risbg-01.ll new file mode 100644 index 00000000000..178fa4f2166 --- /dev/null +++ b/test/CodeGen/SystemZ/risbg-01.ll @@ -0,0 +1,266 @@ +; Test sequences that can use RISBG. +; +; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s + +; Test an extraction of bit 0 from a right-shifted value. +define i32 @f1(i32 %foo) { +; CHECK: f1: +; CHECK: risbg %r2, %r2, 63, 191, 54 +; CHECK: br %r14 + %shr = lshr i32 %foo, 10 + %and = and i32 %shr, 1 + ret i32 %and +} + +; ...and again with i64. +define i64 @f2(i64 %foo) { +; CHECK: f2: +; CHECK: risbg %r2, %r2, 63, 191, 54 +; CHECK: br %r14 + %shr = lshr i64 %foo, 10 + %and = and i64 %shr, 1 + ret i64 %and +} + +; Test an extraction of other bits from a right-shifted value. +define i32 @f3(i32 %foo) { +; CHECK: f3: +; CHECK: risbg %r2, %r2, 60, 189, 42 +; CHECK: br %r14 + %shr = lshr i32 %foo, 22 + %and = and i32 %shr, 12 + ret i32 %and +} + +; ...and again with i64. +define i64 @f4(i64 %foo) { +; CHECK: f4: +; CHECK: risbg %r2, %r2, 60, 189, 42 +; CHECK: br %r14 + %shr = lshr i64 %foo, 22 + %and = and i64 %shr, 12 + ret i64 %and +} + +; Test an extraction of most bits from a right-shifted value. +; The range should be reduced to exclude the zeroed high bits. +define i32 @f5(i32 %foo) { +; CHECK: f5: +; CHECK: risbg %r2, %r2, 34, 188, 62 +; CHECK: br %r14 + %shr = lshr i32 %foo, 2 + %and = and i32 %shr, -8 + ret i32 %and +} + +; ...and again with i64. +define i64 @f6(i64 %foo) { +; CHECK: f6: +; CHECK: risbg %r2, %r2, 2, 188, 62 +; CHECK: br %r14 + %shr = lshr i64 %foo, 2 + %and = and i64 %shr, -8 + ret i64 %and +} + +; Try the next value up (mask ....1111001). The mask itself is suitable +; for RISBG, but the shift is still needed. +define i32 @f7(i32 %foo) { +; CHECK: f7: +; CHECK: srl %r2, 2 +; CHECK: risbg %r2, %r2, 63, 188, 0 +; CHECK: br %r14 + %shr = lshr i32 %foo, 2 + %and = and i32 %shr, -7 + ret i32 %and +} + +; ...and again with i64. +define i64 @f8(i64 %foo) { +; CHECK: f8: +; CHECK: srlg [[REG:%r[0-5]]], %r2, 2 +; CHECK: risbg %r2, [[REG]], 63, 188, 0 +; CHECK: br %r14 + %shr = lshr i64 %foo, 2 + %and = and i64 %shr, -7 + ret i64 %and +} + +; Test an extraction of bits from a left-shifted value. The range should +; be reduced to exclude the zeroed low bits. +define i32 @f9(i32 %foo) { +; CHECK: f9: +; CHECK: risbg %r2, %r2, 56, 189, 2 +; CHECK: br %r14 + %shr = shl i32 %foo, 2 + %and = and i32 %shr, 255 + ret i32 %and +} + +; ...and again with i64. +define i64 @f10(i64 %foo) { +; CHECK: f10: +; CHECK: risbg %r2, %r2, 56, 189, 2 +; CHECK: br %r14 + %shr = shl i64 %foo, 2 + %and = and i64 %shr, 255 + ret i64 %and +} + +; Try a wrap-around mask (mask ....111100001111). The mask itself is suitable +; for RISBG, but the shift is still needed. +define i32 @f11(i32 %foo) { +; CHECK: f11: +; CHECK: sll %r2, 2 +; CHECK: risbg %r2, %r2, 60, 183, 0 +; CHECK: br %r14 + %shr = shl i32 %foo, 2 + %and = and i32 %shr, -241 + ret i32 %and +} + +; ...and again with i64. +define i64 @f12(i64 %foo) { +; CHECK: f12: +; CHECK: sllg [[REG:%r[0-5]]], %r2, 2 +; CHECK: risbg %r2, [[REG]], 60, 183, 0 +; CHECK: br %r14 + %shr = shl i64 %foo, 2 + %and = and i64 %shr, -241 + ret i64 %and +} + +; Test an extraction from a rotated value, no mask wraparound. +; This is equivalent to the lshr case, because the bits from the +; shl are not used. +define i32 @f13(i32 %foo) { +; CHECK: f13: +; CHECK: risbg %r2, %r2, 56, 188, 46 +; CHECK: br %r14 + %parta = shl i32 %foo, 14 + %partb = lshr i32 %foo, 18 + %rotl = or i32 %parta, %partb + %and = and i32 %rotl, 248 + ret i32 %and +} + +; ...and again with i64. +define i64 @f14(i64 %foo) { +; CHECK: f14: +; CHECK: risbg %r2, %r2, 56, 188, 14 +; CHECK: br %r14 + %parta = shl i64 %foo, 14 + %partb = lshr i64 %foo, 50 + %rotl = or i64 %parta, %partb + %and = and i64 %rotl, 248 + ret i64 %and +} + +; Try a case in which only the bits from the shl are used. +define i32 @f15(i32 %foo) { +; CHECK: f15: +; CHECK: risbg %r2, %r2, 47, 177, 14 +; CHECK: br %r14 + %parta = shl i32 %foo, 14 + %partb = lshr i32 %foo, 18 + %rotl = or i32 %parta, %partb + %and = and i32 %rotl, 114688 + ret i32 %and +} + +; ...and again with i64. +define i64 @f16(i64 %foo) { +; CHECK: f16: +; CHECK: risbg %r2, %r2, 47, 177, 14 +; CHECK: br %r14 + %parta = shl i64 %foo, 14 + %partb = lshr i64 %foo, 50 + %rotl = or i64 %parta, %partb + %and = and i64 %rotl, 114688 + ret i64 %and +} + +; Test a 32-bit rotate in which both parts of the OR are needed. +; This needs a separate shift (although RISBLG would be better +; if supported). +define i32 @f17(i32 %foo) { +; CHECK: f17: +; CHECK: rll [[REG:%r[0-5]]], %r2, 4 +; CHECK: risbg %r2, [[REG]], 57, 190, 0 +; CHECK: br %r14 + %parta = shl i32 %foo, 4 + %partb = lshr i32 %foo, 28 + %rotl = or i32 %parta, %partb + %and = and i32 %rotl, 126 + ret i32 %and +} + +; ...and for i64, where RISBG should do the rotate too. +define i64 @f18(i64 %foo) { +; CHECK: f18: +; CHECK: risbg %r2, %r2, 57, 190, 4 +; CHECK: br %r14 + %parta = shl i64 %foo, 4 + %partb = lshr i64 %foo, 60 + %rotl = or i64 %parta, %partb + %and = and i64 %rotl, 126 + ret i64 %and +} + +; Test an arithmetic shift right in which some of the sign bits are kept. +; The SRA is still needed. +define i32 @f19(i32 %foo) { +; CHECK: f19: +; CHECK: sra %r2, 28 +; CHECK: risbg %r2, %r2, 59, 190, 0 +; CHECK: br %r14 + %shr = ashr i32 %foo, 28 + %and = and i32 %shr, 30 + ret i32 %and +} + +; ...and again with i64. +define i64 @f20(i64 %foo) { +; CHECK: f20: +; CHECK: srag [[REG:%r[0-5]]], %r2, 60 +; CHECK: risbg %r2, [[REG]], 59, 190, 0 +; CHECK: br %r14 + %shr = ashr i64 %foo, 60 + %and = and i64 %shr, 30 + ret i64 %and +} + +; Now try an arithmetic right shift in which the sign bits aren't needed. +; Introduce a second use of %shr so that the ashr doesn't decompose to +; an lshr. +define i32 @f21(i32 %foo, i32 *%dest) { +; CHECK: f21: +; CHECK: risbg %r2, %r2, 60, 190, 36 +; CHECK: br %r14 + %shr = ashr i32 %foo, 28 + store i32 %shr, i32 *%dest + %and = and i32 %shr, 14 + ret i32 %and +} + +; ...and again with i64. +define i64 @f22(i64 %foo, i64 *%dest) { +; CHECK: f22: +; CHECK: risbg %r2, %r2, 60, 190, 4 +; CHECK: br %r14 + %shr = ashr i64 %foo, 60 + store i64 %shr, i64 *%dest + %and = and i64 %shr, 14 + ret i64 %and +} + +; Check that we use RISBG for shifted values even if the AND is a +; natural zero extension. +define i64 @f23(i64 %foo) { +; CHECK: f23: +; CHECK: risbg %r2, %r2, 56, 191, 62 +; CHECK: br %r14 + %shr = lshr i64 %foo, 2 + %and = and i64 %shr, 255 + ret i64 %and +}