148 lines · plain
1; Test integer absolute.2;3; RUN: llc < %s -mtriple=s390x-linux-gnu | FileCheck %s4 5; Test i32->i32 absolute using slt.6define i32 @f1(i32 %val) {7; CHECK-LABEL: f1:8; CHECK: lpr %r2, %r29; CHECK: br %r1410 %cmp = icmp slt i32 %val, 011 %neg = sub i32 0, %val12 %res = select i1 %cmp, i32 %neg, i32 %val13 ret i32 %res14}15 16; Test i32->i32 absolute using sle.17define i32 @f2(i32 %val) {18; CHECK-LABEL: f2:19; CHECK: lpr %r2, %r220; CHECK: br %r1421 %cmp = icmp sle i32 %val, 022 %neg = sub i32 0, %val23 %res = select i1 %cmp, i32 %neg, i32 %val24 ret i32 %res25}26 27; Test i32->i32 absolute using sgt.28define i32 @f3(i32 %val) {29; CHECK-LABEL: f3:30; CHECK: lpr %r2, %r231; CHECK: br %r1432 %cmp = icmp sgt i32 %val, 033 %neg = sub i32 0, %val34 %res = select i1 %cmp, i32 %val, i32 %neg35 ret i32 %res36}37 38; Test i32->i32 absolute using sge.39define i32 @f4(i32 %val) {40; CHECK-LABEL: f4:41; CHECK: lpr %r2, %r242; CHECK: br %r1443 %cmp = icmp sge i32 %val, 044 %neg = sub i32 0, %val45 %res = select i1 %cmp, i32 %val, i32 %neg46 ret i32 %res47}48 49; Test i32->i64 absolute.50define i64 @f5(i32 %val) {51; CHECK-LABEL: f5:52; CHECK: lpgfr %r2, %r253; CHECK: br %r1454 %ext = sext i32 %val to i6455 %cmp = icmp slt i64 %ext, 056 %neg = sub i64 0, %ext57 %res = select i1 %cmp, i64 %neg, i64 %ext58 ret i64 %res59}60 61; Test i32->i64 absolute that uses an "in-register" form of sign extension.62define i64 @f6(i64 %val) {63; CHECK-LABEL: f6:64; CHECK: lpgfr %r2, %r265; CHECK: br %r1466 %trunc = trunc i64 %val to i3267 %ext = sext i32 %trunc to i6468 %cmp = icmp slt i64 %ext, 069 %neg = sub i64 0, %ext70 %res = select i1 %cmp, i64 %neg, i64 %ext71 ret i64 %res72}73 74; Test i64 absolute.75define i64 @f7(i64 %val) {76; CHECK-LABEL: f7:77; CHECK: lpgr %r2, %r278; CHECK: br %r1479 %cmp = icmp slt i64 %val, 080 %neg = sub i64 0, %val81 %res = select i1 %cmp, i64 %neg, i64 %val82 ret i64 %res83}84 85; Test another form of f6, which is that produced by InstCombine.86define i64 @f8(i64 %val) {87; CHECK-LABEL: f8:88; CHECK: lpgfr %r2, %r289; CHECK: br %r1490 %shl = shl i64 %val, 3291 %ashr = ashr i64 %shl, 3292 %neg = sub i64 0, %ashr93 %cmp = icmp slt i64 %shl, 094 %abs = select i1 %cmp, i64 %neg, i64 %ashr95 ret i64 %abs96}97 98; Try again with sle rather than slt.99define i64 @f9(i64 %val) {100; CHECK-LABEL: f9:101; CHECK: lpgfr %r2, %r2102; CHECK: br %r14103 %shl = shl i64 %val, 32104 %ashr = ashr i64 %shl, 32105 %neg = sub i64 0, %ashr106 %cmp = icmp sle i64 %shl, 0107 %abs = select i1 %cmp, i64 %neg, i64 %ashr108 ret i64 %abs109}110 111; Repeat f8 with the operands reversed.112define i64 @f10(i64 %val) {113; CHECK-LABEL: f10:114; CHECK: lpgfr %r2, %r2115; CHECK: br %r14116 %shl = shl i64 %val, 32117 %ashr = ashr i64 %shl, 32118 %neg = sub i64 0, %ashr119 %cmp = icmp sgt i64 %shl, 0120 %abs = select i1 %cmp, i64 %ashr, i64 %neg121 ret i64 %abs122}123 124; Try again with sge rather than sgt.125define i64 @f11(i64 %val) {126; CHECK-LABEL: f11:127; CHECK: lpgfr %r2, %r2128; CHECK: br %r14129 %shl = shl i64 %val, 32130 %ashr = ashr i64 %shl, 32131 %neg = sub i64 0, %ashr132 %cmp = icmp sge i64 %shl, 0133 %abs = select i1 %cmp, i64 %ashr, i64 %neg134 ret i64 %abs135}136 137; Repeat f5 with the comparison on the unextended value.138define i64 @f12(i32 %val) {139; CHECK-LABEL: f12:140; CHECK: lpgfr %r2, %r2141; CHECK: br %r14142 %ext = sext i32 %val to i64143 %cmp = icmp slt i32 %val, 0144 %neg = sub i64 0, %ext145 %abs = select i1 %cmp, i64 %neg, i64 %ext146 ret i64 %abs147}148